python – 使用m个可能的元素生成长度为n的列表

我需要在
Python中生成大量的列表.每个列表的长度为13,我有4个可能的值可以进入每个元素.这些是[1,-1,i,-i],但它可以是任何东西.

因此,考虑到主题中的信息,我应该得到4 * 4 * 4 … * 4 = 4 ^ 13 = 67,108,864个列表,或者更一般地,m ^ n.

我在Python的itertools中尝试了combination_with_replacement方法,但是使用以下代码我只得到560个结果.

c = it.combinations_with_replacement([1,-1,np.complex(0,1), np.complex(0,-1)], 13)
print list(c)

我知道组合不关心顺序,所以这个结果可能是正确的.但是,当我使用置换方法时,我只能选择第二个参数< =第一个参数中的元素个数. 知道怎么做到这一点? 谢谢!

最佳答案 我想你想要的

y = itertools.product((1, -1, 1j, -1j), repeat=13)

然后,顺便说一句,打印总和(1表示x中的y)打印,67108864,如您所料.

点赞