使用Python为多个列表压缩未知数量的列表

时间:2016-08-29 20:38:42

标签: python list zip

我需要做一些与此处How would you zip an unknown number of lists in Python?非常相似的事情,但在更一般的情况下。

我有以下设置:

a = [['a', '0', 'b', 'f', '78']]
b = [['3', 'w', 'hh', '12', '8']]
c = [['g', '7', '1', 'a0', '9'], ['45', '4', 'fe', 'h', 'k']]

我需要将这些列表压缩在一起以获得:

abc =  [['a', '3', 'g', '45'], ['0', 'w', '7', '4'], ['b', 'hh', '1', 'fe'], ['f', '12', 'a0', 'h'], ['78', '8', '9', 'k']]

我可以用:

生成
zip(a[0], b[0], c[0], c[1])

但是列表a,b,c包含许多子列表,这些子列表会因连续运行而有所不同,因此这种扩展它们的“手动”方式将无效。

我能得到的最接近的是:

zip(a[0], b[0], *c)

由于不允许在除最后一个位置之外的任何其他位置解压缩包含*的列表,因此“理想”表达式:

zip(*a, *b, *c)

不起作用。

如何将多个列表与未知数量的子列表压缩在一起?

1 个答案:

答案 0 :(得分:5)

itertools.chain救援:

import itertools

z = list(zip(*itertools.chain(a, b, c)))