如何在NumPy中沿新轴将一维数组与二维数组组合

你可以使用numpy.stack()函数，指定axis=1参数来实现需求。具体步骤如下：

1. 先将输入的列表转换为NumPy数组：

import numpy as np

a = np.array([[1,2,3], [1,3,5], [1,1,1]])
b = np.array([[1], [2], [5]])

2. 使用np.stack()沿第二个维度（axis=1）堆叠两个数组：

result = np.stack([a, b], axis=1)

执行后得到的结果就是你需要的结构：

array([[[1, 2, 3],
        [1]],

       [[1, 3, 5],
        [2]],

       [[1, 1, 1],
        [5]]])

补充说明 #

np.stack()会在指定的轴上创建新维度，这里axis=1表示将两个数组的对应行进行配对，最终生成形状为(3, 2)的数组，其中每个位置的元素分别是原数组a的一行和数组b的对应行。

内容的提问来源于stack exchange，提问作者MagicBBanana