加入收藏 | 设为首页 | 会员中心 | 我要投稿 李大同 (https://www.lidatong.com.cn/)- 科技、建站、经验、云计算、5G、大数据,站长网!
当前位置: 首页 > 编程开发 > Python > 正文

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

发布时间:2020-12-20 11:43:46 所属栏目:Python 来源:网络整理
导读:例如,如果我有一个选择集K. K = ['a','b','c'] 和长度N. N = 4 我想尽可能地回报: ['a','a','a']['a','b']['a','c']['a','a']...['c','c','c'] 我可以用递归来做,但它没有意思.有更多的Pythonic方式吗? 解决方法 这可以用 itertools 完成. K = ['a','c'] i
例如,如果我有一个选择集K.

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

和长度N.

N = 4

我想尽可能地回报:

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

我可以用递归来做,但它没有意思.有更多的Pythonic方式吗?

解决方法

这可以用 itertools完成.

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

编辑:我意识到你实际上想要product,而不是combinations_with_replacement.更新的代码.

(编辑:李大同)

【声明】本站内容均来自网络,其相关言论仅代表作者个人观点,不代表本站立场。若无意侵犯到您的权利,请及时与联系站长删除相关内容!

    推荐文章
      热点阅读