用于筛选的生成器函数
从输入的可迭代对象中产生项的子集,而且不修改项本身。多数可筛选生成器接收一个predicate参数。这个参数的值是个布尔函数,接收一个参数,应用到输入中的每一项上,用于判断项是否包含在输出中。
itertools.compress(it,selectors_it)
并行处理两个可迭代对象;如果selectors_it的值为真值,那么产生iter对应的项。
it = "床前明月光疑是地上霜"
selectors_it = [1, 0, True, False, '', 1, 1, 0, 1, None]
res = itertools.compress(it, selectors_it)
print(list(res)) # 输出['床', '明', '疑', '是', '上']
itertools.dropwhile(predicate,it)
处理it,跳过predicate计算结果为真值的项,然后产生剩下的项。
numbers = [3, 5, 7, 10, 12, 13, 14, 15]
# 弃掉所有小于10的元素
droped_numbers = itertools.dropwhile(lambda x: x < 10, numbers)
print(list(droped_numbers)) # [10, 12, 13, 14, 15]