python过滤工具

过滤工具

itertools.compress()以一个iterable对象和相应的Boolean选择器序列作为输入，然后输出iterable对象中对应选择器为True的元素，得到一个迭代器（可以使用list来转换为列表）。它能够十分方便的根据一个序列过滤另一个序列。

from itertools import compress
counts = [1, 2, 3, 4, 2, 8, 6, 4]
data = ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h']
selector = [n > 5 for n in counts]
print(list(compress(data, selector)))

自己怎么实现

def compress(data,selector):
    return (d for d,s in zip(data,selector) if s)