Python-从长度不等的列表中获取所有唯一的组合并进行替换

注意：这不是标题可能说的重复问题

如果我有一个 list 列表，我需要通过替换从中获取所有组合。

import itertools

l = [[1,2,3] ,[1,2,3],  [1,2,3]]
n = []
for i in itertools.product(*l):
    if sorted(i) not in n:
        n.append(sorted(i))
for i in n:
    print(i)

[1, 1, 1]
[1, 1, 2]
[1, 1, 3]
[1, 2, 2]
[1, 2, 3]
[1, 3, 3]
[2, 2, 2]
[2, 2, 3]
[2, 3, 3]
[3, 3, 3]

感谢@RoadRunner 和@Idlehands。

上面的代码很完美，但有两个问题：

1. 对于大型列表，itertools.product 会抛出 MemoryError。当 l 有 18 个 3 长度的子列表时，可提供约 4 亿个组合。

2. 顺序很重要，因此sorted对我的问题不起作用。这可能会让一些人感到困惑，因此用下面的例子进行解释。

   l = [[1,2,3], [1], [1,2,3]]

这里我有两个独特的组：

第 1 组：具有相同值 [1,2,3] 的元素 0, 2

第 2 组：具有值 [1] 的元素 1

因此，我需要的解决方案是：

[1,1,1]
[1,1,2]
[1,1,3]
[2,1,2]
[2,1,3]
[3,1,3]

因此位置1被固定为1.

希望这个例子有帮助。

如何使用 a 以不同的顺序对具有相同元素的序列进行分组？collections.defaultdict https://docs.python.org/3.6/library/collections.html#collections.defaultdict，然后从每个键中选取第一个元素：

from itertools import product
from collections import defaultdict

l = [[1] ,[1,2,3],  [1,2,3]]

d = defaultdict(list)
for x in product(*l):
    d[tuple(sorted(x))].append(x)

print([x[0] for x in d.values()])

这使：

[(1, 1, 1), (1, 1, 2), (1, 1, 3), (1, 2, 2), (1, 2, 3), (1, 3, 3)]

或者，也可以通过保留一组已添加的内容来完成此操作：

from itertools import product

l = [[1] ,[1,2,3],  [1,2,3]]

seen = set()
combs = []

for x in product(*l):
    curr = tuple(sorted(x))
    if curr not in seen:
        combs.append(x)
        seen.add(curr)

print(combs)
# [(1, 1, 1), (1, 1, 2), (1, 1, 3), (1, 2, 2), (1, 2, 3), (1, 3, 3)]

如果您不想排序，请考虑使用frozenset https://docs.python.org/3/library/stdtypes.html?highlight=frozenset#frozenset with collections.Counter() https://docs.python.org/3.6/library/collections.html#collections.Counter:

from collections import Counter
from itertools import product

l = [[1] ,[1,2,3],  [1,2,3]]

seen = set()
combs = []

for x in product(*l):
    curr = frozenset(Counter(x).items())

    if curr not in seen:
        seen.add(curr)
        combs.append(x)

print(combs)
# [(1, 1, 1), (1, 1, 2), (1, 1, 3), (1, 2, 2), (1, 2, 3), (1, 3, 3)]

Note:您还可以使用setdefault()对于第一种方法，如果您不想使用defaultdict().