使用 itertools 来降低四个“for”的复杂性斯蒂姆？

Question

我有 4 个列表：

list1, list2, list3, list4 = range(0,4), range(4,9), range(8,15), range(2,16)
# just using range() generate sample data  

我可以通过以下代码按序列 list1_item、list2_item、list3_item、list4_item 列出所有项目组合：

for item1 in list1:
    for item2 in list2:
        for item3 in list3:
             for item4 in list4:
                   print [item1, item2, item3, item4]

我是否可以通过 itertools 模块？

谢谢

Answer 1

是的，您可以：（

for n in itertools.product(list1, list2, list3, list4):
    print n

n 将是一个元组而不是列表）。

Answer 2

我认为你想要的是 itertools.product

import itertools
ret = list ( itertools.product( list1, list2, list3, list4 ) )

而 ret 就是结果。