如何将常量值添加到 2d numpy 数组的轴 1(仅)?

How to add constant value to axis 1 (only) of 2d numpy array?

假设我有一个二维数组,比如

a = array([[0,1],[10,12]])

是否有一种简单快速(对于大型数组)的方法来创建一个新的二维数组 aplus1,它只向第二个轴添加一个常数,同时保持第一个轴不变?因此,我希望获得 (for constant = 1)

array([[0,1],[11,13]])

我希望 aplus1 = np.add(a,1,axis=1) 可用,但 np.add 没有选项 axis=

IIUC,你可以通过索引来做到这一点:

a[row_index, :] += constant_value

因此,要修改 a

a[1, :] += 1             # [[ 0  1] [11 13]]

或者如果你需要一个新数组,你可以先复制a。所以,例如,如果 row_index = 1constant_value = 2:

b = np.copy(a)
b[1, :] += 2             # [[ 0  1] [12 14]]