Question

总蟒蛇新秀在这里。我希望迭代三个列表，其中前两个应该以所有可能的方式重新组合，而只有一个值附加到第三个列表的每个组合。到目前为止，我得到了：

import itertools

list1 = [1,2,3]
list2 = [1,2,3]
list3 = [1,2,3,4,5,6,7,8,9]

for x, y, in itertools.product(list1, list2):
    print x, y, list3

返回：

1 1 [1, 2, 3, 4, 5, 6, 7, 8, 9]
1 2 [1, 2, 3, 4, 5, 6, 7, 8, 9]
1 3 [1, 2, 3, 4, 5, 6, 7, 8, 9]
2 1 [1, 2, 3, 4, 5, 6, 7, 8, 9]
2 2 [1, 2, 3, 4, 5, 6, 7, 8, 9]
2 3 [1, 2, 3, 4, 5, 6, 7, 8, 9]
3 1 [1, 2, 3, 4, 5, 6, 7, 8, 9]
3 2 [1, 2, 3, 4, 5, 6, 7, 8, 9]
3 3 [1, 2, 3, 4, 5, 6, 7, 8, 9]

但是，我希望它返回以下内容：

我怎样才能得到这个结果？

Answer 1

通过并行迭代两个集合通常使用zip：

完成

import itertools

list1 = [1,2,3]
list2 = [1,2,3]
list3 = [1,2,3,4,5,6,7,8,9]

for (x, y), z in zip(itertools.product(list1, list2), list3):
    print x, y, z

结果：

或者，如果list3总是[1,2,3 ...某些int]，您可以完全丢弃它，只使用enumerate。

import itertools

list1 = [1,2,3]
list2 = [1,2,3]

for idx, (x, y) in enumerate(itertools.product(list1, list2), 1):
    print x, y, idx

如何在python中迭代三个列表

1 个答案: