python - 按列减去 Numpy 数组

Question

所以我有一个示例数组，比如:

import numpy as np
np.array([[[ 85, 723]],
          [[ 86, 722]],
          [[ 87, 722]],
          [[ 89, 724]],
          [[ 88, 725]],
          [[ 87, 725]]])

我想做的是仅从第二列中减去一个数字，例如 10。我希望输出看起来像这样:

np.array([[[ 85, 713]],
          [[ 86, 712]],
          [[ 87, 712]],
          [[ 89, 714]],
          [[ 88, 715]],
          [[ 87, 715]]])

我试过使用 np.subtract，但它不支持沿轴减法(至少据我所知)。

Answer 1

切片和减法-

a[...,1] -= 10

这适用于从第二列中减去任意维数的数组。

sample 运行-

In [582]: a
Out[582]: 
array([[[30, 23]],

       [[36, 88]],

       [[27, 15]],

       [[38, 61]],

       [[79, 14]]])

In [583]: a[...,1] -= 10

In [584]: a
Out[584]: 
array([[[30, 13]],

       [[36, 78]],

       [[27,  5]],

       [[38, 51]],

       [[79,  4]]])