我有一个类似于[{0, 1}, {1, 2}, {2, 3}, {3, 4}, {4, 5}]
的集合列表,我需要计算这些集合之间的乘积,但只保留不产生重复数的解。我找到了一个使用itertools.product
的解决方案,但这仍然计算所有的产品,我想知道是否有一个内置的方法可以立即产生我想要的结果(或者您是否有一个操作数据结构以加快计算时间的建议被广泛接受)。
我的解决方案:
from itertools import product
def pick(sets):
for p in product(*sets):
if len(set(p)) == len(p):
yield p
从而得到所需的输出:
>>> x = [{i, i+1} for i in range(5)]
>>> for p in pick(x):
... print(p)
...
(0, 1, 2, 3, 4)
(0, 1, 2, 3, 5)
(0, 1, 2, 4, 5)
(0, 1, 3, 4, 5)
(0, 2, 3, 4, 5)
(1, 2, 3, 4, 5)
最佳答案
def pick(sets, excluding=frozenset()):
if not sets:
yield ()
return
for x in sets[0] - excluding:
for rest in pick(sets[1:], excluding | {x}):
yield (x,) + rest
关于python - 计算组之间的产品,仅保留没有重复项目的产品,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/45769967/