我想使用itertools.product
或类似的命令来获取迭代器n
副本的笛卡尔乘积,iter
自身,其中n
是一个变量。我尝试通过以下方式递归构造它
import itertools
prod_iter = iter
for i in xrange(n-1):
prod_iter = itertools.product(prod_iter,iter)
但是从,比如说,iter=xrange(2)
开始,n=3
和跑步
for i in prod_iter:
print i
我得到作为输出
((0,0),0)
((0,0),1)
((0,1),0)
...
而不是
(0,0,0)
(0,0,1)
(0,1,0)
...
如我所愿。有没有办法做到这一点?
itertools.product
有这样的参数:repeat
。
所以没有必要自己发明它,你可以简单地写:
from itertools import product
for i in product(iter, repeat=3):
print i
例如:
>>> from itertools import product
>>>
>>> iter = "qux"
>>>
>>> for i in product(iter, repeat=3):
... print i
...
('q', 'q', 'q')
('q', 'q', 'u')
('q', 'q', 'x')
('q', 'u', 'q')
('q', 'u', 'u')
('q', 'u', 'x')
('q', 'x', 'q')
('q', 'x', 'u')
('q', 'x', 'x')
('u', 'q', 'q')
('u', 'q', 'u')
('u', 'q', 'x')
('u', 'u', 'q')
('u', 'u', 'u')
('u', 'u', 'x')
('u', 'x', 'q')
('u', 'x', 'u')
('u', 'x', 'x')
('x', 'q', 'q')
('x', 'q', 'u')
('x', 'q', 'x')
('x', 'u', 'q')
('x', 'u', 'u')
('x', 'u', 'x')
('x', 'x', 'q')
('x', 'x', 'u')
('x', 'x', 'x')