Python - 如何生成大小大于列表元素数的排列

时间:2015-06-05 17:45:11

标签: python list iteration permutation

很抱歉,如果这个标题毫无意义。我的意思是这样的:我有一个列表[A, B, C],我想要那些填充长度为10的列表的所有可能的排列。

[A, B, C] => [A, A, A, A, A, A, A, A, A, A]
          => [A, A, A, A, A, A, A, A, A, B]
          ...
          => [C, C, C, C, C, C, C, C, C, C]

我一直在阅读itertools文档,但permutations函数在这种情况下不起作用,除非输出列表长度小于或等于3.谢谢!

1 个答案:

答案 0 :(得分:4)

您正在制作值的产品,因此请使用带有重复集的itertools.product()

from itertools import product

for combo in product(['A', 'B', 'C'], repeat=10):

演示:

>>> from itertools import product
>>> products = product(['A', 'B', 'C'], repeat=10)
>>> next(products)
('A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A')
>>> next(products)
('A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'B')
>>> next(products)
('A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'C')
>>> from itertools import islice
>>> skip_to_end = islice(products, (3 ** 10) - 6, None)
>>> next(skip_to_end)
('C', 'C', 'C', 'C', 'C', 'C', 'C', 'C', 'C', 'A')
>>> next(skip_to_end)
('C', 'C', 'C', 'C', 'C', 'C', 'C', 'C', 'C', 'B')
>>> next(skip_to_end)
('C', 'C', 'C', 'C', 'C', 'C', 'C', 'C', 'C', 'C')