Python itertools-takewhile():多个谓词

Question

假设一个发生器逐个产生以下元组(从左到右)

(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).)

Answer 1

您可以使用索引访问每个元组的元素.

list(itertools.takewhile(lambda x: sum(x) < 20 and x[0] > 3, some_generator()))