如何将常量值添加到 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 = 1
和 constant_value = 2
:
b = np.copy(a)
b[1, :] += 2 # [[ 0 1] [12 14]]
假设我有一个二维数组,比如
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 = 1
和 constant_value = 2
:
b = np.copy(a)
b[1, :] += 2 # [[ 0 1] [12 14]]