[英]How can I lazily generate and process combinations of results from a generator?
我有一个生成器,我想在它上面执行一个嵌套循环,这样内部循环将从当前外部循环所在的位置开始。 例如,我有一个生成列表[1,2,3]
的生成器,我的循环应该生成: (1,2),(1,3),(2,3)
。 我想出的代码如下:
from itertools import tee
def my_gen():
my_list = [1, 2, 3]
for x in my_list:
yield x
first_it = my_gen()
while True:
try:
a = next(first_it)
first_it, second_it = tee(first_it)
for b in second_it:
print(a,b)
except StopIteration:
break
这段代码很麻烦,效率不高,对我来说看起来不是很pythonic。 请注意,我不能使用combinations_with_replacement
,因为我需要一个内部循环来处理来自外部循环的特定值。
对更优雅和 Pythonic 代码有什么建议吗?
只重复克隆和耗尽生成的迭代器之一不是很有效。 根据itertools.tee
文档:
一般来说,如果一个迭代器在另一个迭代器启动之前使用了大部分或全部数据,那么使用 list() 而不是 tee() 会更快。
from itertools import islice
my_list = [1, 2, 3]
# or, more generally
# my_list = list(my_gen())
for i, a in enumerate(my_list):
for b in islice(my_list, i+1, None):
print((a, b))
(1, 2)
(1, 3)
(2, 3)
结果(1,2),(1,3),(2,3)
不是笛卡尔积。 您的意思是无需重复即可获得所有组合?
对于没有重复的组合
使用itertools.combinations
函数:
from itertools import combinations
print(list(combinations([1,2,3], 2)))
对于实际的笛卡尔积
使用列表推导,您可以将代码简化为更 Pythonic 和优雅的方式:
my_list = [1, 2, 3]
c_prod = [(i, j) for i in iter(my_list) for j in iter(my_list)]
print(c_prod)
另一种选择是使用itertools.product
函数:
import itertools
for i in itertools.product(my_list, my_list):
print(i)
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.