python - 如何使用 numpy 进行按列减法？

Question

假设我有:

array([[ 5,  4,  3,  3],
       [ 5,  4,  3,  3],
       [ 5,  4,  3,  3],
       [ 5,  4,  3,  3],
       [ 5,  4,  3,  3]])

然后我就有了

数组([1,2,3,4])

我想用一维数组中相应的列值减去原始矩阵中的每一列。

所以，我希望它成为

array([[ 4,  2,  0,  -1],
       [ 4,  2,  0,  -1],
       [ 4,  2,  0,  -1],
       [ 4,  2,  0,  -1],
       [ 4,  2,  0,  -1]])

如何在 numpy 中实现这一点？

Answer 1

这比您想象的要容易得多:

In [1]: import numpy as np

In [2]: arr = np.array([[ 5,  4,  3,  3],
    ...:                [ 5,  4,  3,  3],
    ...:                [ 5,  4,  3,  3],
    ...:                [ 5,  4,  3,  3],
    ...:                [ 5,  4,  3,  3]])

In [3]: sub = np.array([1, 2, 3, 4])

In [4]: arr - sub
Out[4]:
array([[ 4,  2,  0, -1],
       [ 4,  2,  0, -1],
       [ 4,  2,  0, -1],
       [ 4,  2,  0, -1],
       [ 4,  2,  0, -1]])

NumPy 会自动广播，因此只需使用 - 运算符即可!