python - 在 python 中合并一维和二维列表

Question

我正在尝试设置数据以转换为 numpy 数组。我有三个 list 。两个是一维的，一个是二维的。

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()，但它没有深入研究二维数组。

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 列表。