python - 在 Python 中将多个列表合并为一个列表

Question

我有一个列表 X。我想附加元素 X[0][0],X[1][0],X[2][0], X[1][0],X[ 1][1],X[2][1] 等等。我展示了当前和预期的产出。

X=[[[1], [2], [3], [4], [5]], [[8], [4], [6], [2], [3]],[[9], [1], [2], [5], [1]]]
X1=[]
for i in range(0,len(X)):
    for j in range(0,len(X[0])):
        X1=X[i][j]
        X1.append(X)

当前输出为

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

预期的输出是

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

Answer 1

使用zip遍历子列表，然后添加每个内部列表，你可以使用列表理解:

>>> [x+y+z for (x,y,z) in zip(X[0], X[1], X[2])]

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

或者你可以使用 functools 中的 reduce 并将解压后的列表传递给 zip 如果你不确定有多少子列表

>>> from functools import reduce 
>>> [reduce(lambda x,y: x+y, v) for v in zip(*X)]

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

另一种方法是调用 sum 迭代传递给 zip 的解压缩列表，正如@Mad Physicist 在评论中所建议的:

>>> [sum(x, start=[]) for x in zip(*X)]

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