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

2023年1月10日 178次阅读

例如,如果我有一个选择集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方式吗？

最佳答案 这可以用
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.更新的代码.

点赞