从一个列表中获取基于另一个布尔列表的元素

Question

我有两个相同大小的列表，一个是字符串列表，另一个是布尔值列表（True，False），我想只返回一个字符串列表索引是真的。

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

想要

s_list = ['abc','def'] 

Answer 1

使用itertools.compress

  

compress（数据，选择器）：返回与选择器元素对应的数据元素

因此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]