假设一个生成器一个一个地生成下面的元组(从左到右)
(1, 2, 3), (2, 5, 6), (3, 7, 10), (4, 5, 11), (3, 5, 15), (4, 5, 9), (4, 6, 12)
...
并假设只要谓词为真,我就想进行迭代。令该谓词为 sum(yielded_value) < 20
.然后迭代器将停止 (3, 5, 15)
.我可以做到,说:
list(itertools.takewhile(lambda x: sum(x) < 20, some_generator()))
问题,如何写一个带有两个谓词的相似表达式?假设我想要:
list(itertools.takewhile(lambda x: sum(x) < 20 and first_value_of_tuple > 3, some_generator()))
(在这种情况下,停止 (4, 6, 12)
。)
最佳答案
您可以通过索引访问每个元组的元素。
list(itertools.takewhile(lambda x: sum(x) < 20 and x[0] > 3, some_generator()))
关于Python itertools—takewhile() : multiple predicates,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/34187448/