从python中的另一个列表中获取列表的相应元素时,获取错误的列表使用map函数

Question

我有两个清单:

a = [1,2,3,4]
b = [True,False,True,False]

我希望得到的元素a是coresponding到True中b,不使用for循环.我使用该map函数来解决它,但它出错了:

def f(x,y):
    if x:
        return y
s = list(map(f,b,a))

s是的[1, None, 3, None],我不想要None,我该怎么办？

Answer 1

用途compress:

from itertools import compress
a = [1,2,3,4]
b = [True,False,True,False]
s = list(compress(a, b))

输出:

[1, 3]