根据另一个布尔列表从一个列表中获取元素

Question

我有两个大小相同的列表，一个是字符串列表，另一个是布尔值列表（ True ， False ），并且仅当索引为True时，我才想返回字符串列表。

b_list = [True, False, True]
s_list = ['abc', 'sfsfsfsf', 'def']

想

s_list = ['abc','def'] 

Answer 1

使用itertools.compress

compress（data，selectors） ：返回与真正的选择器元素相对应的数据元素

所以s_list是数据 ， b_list是选择器 ：

In [8]: import itertools

In [9]: list(itertools.compress(s_list, b_list))
Out[9]: ['abc', 'def']

Answer 2

如果没有使用列表理解的itertools ：

[y for (x,y) in zip(b_list, s_list) if x]