列表的迭代

Question

from itertools import permutations 
l = [0, 1, 2, 3, 4] 
x = permutations (l, 3) 

每个排列我得到以下几点：

(0, 1, 2) , (0, 1, 3), ...., (0, 2, 1), (0, 2, 3), (0,2,4),...., (4, 3, 0), (4, 3, 1), 
(4, 3, 2) 

这是什么预期。 但我需要的是：

(0, 0, 0), (0, 0, 1), ...., (0, 0, 4), (0, 1, 0), (0, 1, 1)........ 

如何实现这一目标？

Answer 1

你需要的是一个置换与更换，或一个产品，而是itertool的permutations产生排列无需更换。您可以自己计算一下产品：

[(x,y,z) for x in l for y in l for z in l] 
#[(0, 0, 0), (0, 0, 1), (0, 0, 2), (0, 0, 3), (0, 0, 4), (0, 1, 0), ... 

或者从itertools使用同名函数：

list(itertools.product(l,repeat=3)) 
# [(0, 0, 0), (0, 0, 1), (0, 0, 2), (0, 0, 3), (0, 0, 4), (0, 1, 0),... 

后一种方法更有效。

Answer 2

您需要使用product，不使用permutations，从itertools模块这样的例子：

from itertools import product 

l = [0, 1, 2, 3, 4] 
# Or: 
# b = list(product(l, repeat=3)) 
b = list(product(l,l,l)) 
print(b) 

输出：

[(0, 0, 0), (0, 0, 1), (0, 0, 2), (0, 0, 3), ..., (4, 4, 1), (4, 4, 2), (4, 4, 3), (4, 4, 4)] 

Answer 3

您需要的产品，而不是置换

from itertools import product 
l = [0, 1, 2, 3, 4] 
b = list(product(l, repeat=3))