Question

例如，如果我有一个选择集K

K = ['a','b','c']

和长度N

N = 4

我想尽可能地回归：

['a','a','a','a']
['a','a','a','b']
['a','a','a','c']
['a','a','b','a']
...
['c','c','c','c']

我可以用递归来做，但它没有意思。是否有更多的Pythonic方式？

Answer 1

可以使用itertools完成。

>>> K = ['a','b','c']
>>> import itertools
>>> N = 4
>>> i = itertools.product(K,repeat = N)
>>> l = [a for a in i]
>>> l[:3]
[('a', 'a', 'a', 'a'), ('a', 'a', 'a', 'b'), ('a', 'a', 'a', 'c')]

编辑：我意识到你真的想要product，而不是combinations_with_replacement。更新了代码。

获取每个元素的K个选项的N个项目列表？

1 个答案: