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)]

例如

l = list(itertools.product(a,b,c))
for ai, bi, ci in l:
    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?

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

2 个答案: