Python-找到可能的组合从项目和数量的代表[重复]

Question

我有一本字典，上面有项目和每个项目的代表人数，例如：

{A: 3, B: 1}

我正在寻找一种方法来获得所有可能的不同组合，顺序并不重要：[AB]与[BA]相同。

想要的输出是：

[A, AA, AAA, B, AB, AAB, AAAB]

Answer 1

下面是一种使用列表理解和itertools.product的方法：

from itertools import product, chain
d = {'A': 3, 'B': 1}
l = [[k*i for i in range(1,v+1)] for k, v in d.items()]
[''.join(element) for element in product(*l)] + list(chain.from_iterable(l))

输出：

['AB', 'AAB', 'AAAB', 'A', 'AA', 'AAA', 'B']

详情：

第一个列表理解的结果是一个嵌套列表，其中每个子列表都有字典中包含的每个键，从1重复到其相应的值：

[['A', 'AA', 'AAA'], ['B']]

第二个，使用itertools执行笛卡尔乘积。产品，并且每个产品都被连接成一个字符串。最后，使用链添加l的扁平版本。从_iterable：

print(list(chain.from_iterable(l)))
['A', 'AA', 'AAA', 'B']

[''.join(element) for element in product(*l)]
['AB', 'AAB', 'AAAB']