python itertools卡死_Python的itertools.product()效率

所以我正在研究计算n个数组的笛卡尔积的不同方法,并且我遇到了使用以下代码的相当优雅的解决方案(此处为SO)：

import itertools

for array in itertools.product(*arrays):

print array

看一下对于itertools.product()的python doc page(我用2.7,顺便说一句),它说代码等同于以下内容：

def product(*args, **kwds):

# product('ABCD', 'xy') --> Ax Ay Bx By Cx Cy Dx Dy

# product(range(2), repeat=3) --> 000 001 010 011 100 101 110 111

pools = map(tuple, args) * kwds.get('repeat', 1)

result = [[]]

for pool in pools:

result = [x+[y] for x in result for y in pool]

for prod in result:

yield tuple(prod)

(它确实注意到以下内容：此函数等效于以下代码,但实际实现不会在内存中构建中间结果:)

我不是CS人 – 所以我很难估计这个算法的效率.我的第一个猜测是O(n ^ 2)(由于嵌套的for循环).

我错了吗？