向 NumPy 数组添加额外的列

假设您有一个 2D NumPy 数组 a 如下：

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

要沿第二个轴添加一列零，您可以使用各种方法。一种方法是使用 np.c_[ ] 函数：

b = np.c_[a, np.zeros(a.shape[0])]

这将创建一个新的数组 b 附加一列零：

b = np.array([
    [1, 2, 3, 0],
    [2, 3, 4, 0],
])

或者，您可以使用 np.r_[ ] 函数：

b = np.r_[a, np.zeros((a.shape[0], 1))]

此方法还将向数组添加一列零。

请注意 np.r_[ ] 和np.c_[ ] 提供了修改数组维度的灵活选项。它们可用于混合向量和标量、添加行或列，甚至在指定位置插入整个数组。