在python中合并1D和2D列表

Question

我正在尝试设置数据以转换为numpy数组.我有三个清单.两个是一维的,一个是二维的.

a = [1,2,3]
b = [4,5,6]
c = [ [7,8],[9,10],[11,12] ]

我想最终得到这个:

[ [1,4,7,8],[2,5,9,10],[3,6,11,12] ]

我尝试过使用zip(),但它没有深入研究2D数组.

Answer 1

假设您不介意在转换本身中使用 NumPy，则以下内容应该有效。

from numpy import array

a = array([1, 2, 3])
b = array([4, 5, 6])
c = array([[7, 8], [9, 10], [11, 12]])

result = array(list(zip(a, b, c[:, 0], c[:, 1])))

请注意，它c[:, n]仅适用于 NumPy 数组，不适用于标准 Python 列表。