在 Python 中迭代多个列表的项目组合

Question

我有几个列表，我需要对这些列表项的每个可能组合做一些事情。 在两个列表的情况下，我可以这样做：

for a in alist:
  for b in blist:
    # do something with a and b

但是，如果有更多的列表，比如 6 或 7 个列表，这种方法似乎不太情愿。 有没有办法优雅地实现这个迭代？

Answer 1

您可以使用itertools.product从您的列表中进行所有可能的组合。 结果将是一个长长的tuple列表，每个列表中的元素按照您传入列表的顺序排列。

>>> a = [1,2,3]
>>> b = ['a', 'b', 'c']
>>> c = [4,5,6]
>>> import itertools

>>> list(itertools.product(a,b,c))
[(1, 'a', 4), (1, 'a', 5), (1, 'a', 6), (1, 'b', 4), (1, 'b', 5), (1, 'b', 6), (1, 'c', 4), (1, 'c', 5), (1, 'c', 6),
 (2, 'a', 4), (2, 'a', 5), (2, 'a', 6), (2, 'b', 4), (2, 'b', 5), (2, 'b', 6), (2, 'c', 4), (2, 'c', 5), (2, 'c', 6),
(3, 'a', 4), (3, 'a', 5), (3, 'a', 6), (3, 'b', 4), (3, 'b', 5), (3, 'b', 6), (3, 'c', 4), (3, 'c', 5), (3, 'c', 6)]

例如

for ai, bi, ci in itertools.product(a,b,c):
    print ai, bi, ci

输出

1 a 4
1 a 5
1 a 6
... etc

Answer 2

如果实际上有 6 或 7 个列表，为了可读性，可能值得去itertools.product 。 但是对于简单的情况，使用list comprehension很简单，并且不需要导入。 例如：

alist = [1, 2, 3]
blist = ['A', 'B', 'C']
clist = ['.', ',', '?']

abc = [(a,b,c) for a in alist for b in blist for c in clist]

for e in abc:
    print("{}{}{} ".format(e[0],e[1],e[2])),

# 1A.  1A,  1A?  1B.  1B,  1B?  1C.  1C,  1C?  2A.  2A,  2A?  2B.  2B,  2B?  2C.  2C,  2C?  3A.  3A,  3A?  3B.  3B,  3B?  3C.  3C,  3C?