提问人：Kalev Maricq 提问时间：7/19/2020 最后编辑：Kalev Maricq 更新时间：7/20/2020 访问量：228
在 python 中按列连接数组
Concatenate arrays by column in python
问：
我有一个数组列表，其中每个数组都是一个列表列表。我想把它变成一个包含所有列的数组。我尝试使用 for 循环来完成这项工作，但感觉它在列表理解中应该是可行的。有没有一个很好的单行可以做到这一点？
Example Input: [[[1,2],[3,4],[5,6]],[[7,8],[9,10],[11,12]]]
Desired Output: [[1,2,7,8],[3,4,9,10],[5,6,11,12]]
注意：示例在主列表中只有两个数组，但我的实际数据还有更多，所以我正在寻找适用于 N 个子数组的东西。
编辑： 尝试解决此问题的示例
适用于两个，但不能一概而论：
[input[0][i]+input[1][i] for i in range(len(input[0]))]
这些不起作用，但显示了这个想法：
[[element for table in input for element in row] for row in table]
[[*imput[j][i] for j in range(len(input))] for i in range(len(input[0]))]
编辑：仅使用列表推导和zip的选定答案，但所有答案（截至目前）都有效，因此请使用最适合您的风格/用例的答案。
答：
3赞
BluePigeon
7/19/2020
#1
这是一种方法：
initial = [[[1,2],[3,4],[5,6]],[[7,8],[9,10],[11,12]]]
output = [a+b for a, b in zip(*initial)]
print(output)
如果您有更多列表，这也有效：
import itertools
initial = [[[1,2],[3,4],[5,6]],[[7,8],[9,10],[11,12]],[[13,14],[15,16],[17,18]]]
output = [list(itertools.chain.from_iterable(values)) for values in zip(*initial)]
print(output)
评论
1赞
Kalev Maricq
7/19/2020
我喜欢使用zip的想法，我没有想到这一点。这似乎只适用于列表中的两个数组。有没有办法概括它？
5赞
ggorlen
7/19/2020
#2
>>> L = [[[1,2],[3,4],[5,6]],[[7,8],[9,10],[11,12]]]
>>> list([y for z in x for y in z] for x in zip(*L))
[[1, 2, 7, 8], [3, 4, 9, 10], [5, 6, 11, 12]]
>>> L = [[[1,2],[3,4],[5,6]],[[7,8],[9,10],[11,12]],[[13,14],[15,16],[17,18]]]
>>> list([y for z in x for y in z] for x in zip(*L))
[[1, 2, 7, 8, 13, 14], [3, 4, 9, 10, 15, 16], [5, 6, 11, 12, 17, 18]]
3赞
jizhihaoSAMA
7/19/2020
#3
如果你不介意的话，它是列表中的一个元组。您也可以尝试：
from itertools import chain
a = [[[1, 2], [3, 4], [5, 6]], [[7, 8], [9, 10], [11, 12]], [[13, 14], [15, 16], [17, 18]]]
output = list(map(list, map(chain.from_iterable, zip(*a))))
# [[1, 2, 7, 8, 13, 14], [3, 4, 9, 10, 15, 16], [5, 6, 11, 12, 17, 18]]
2赞
Nicolas Gervais
7/19/2020
#4
这样就可以了，我命名了你的输入：first
[*map(lambda x: list(i for s in x for i in s), zip(*first))]
[[1, 2, 7, 8], [3, 4, 9, 10], [5, 6, 11, 12]]
上一个：具有缺失值的加权协方差矩阵
下一个：比较相邻列表项
评论