我需要做一些与此处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)
不起作用。
如何将多个列表与未知数量的子列表压缩在一起?
答案 0 :(得分:5)
itertools.chain
救援:
import itertools
z = list(zip(*itertools.chain(a, b, c)))