Python从每个矩阵行中减去第一个元素

Question

我想从该行的其余元素中减去一行的第一个值，因此对于

import numpy as np

z = np.array([[1,2,3,4],[4,5,6,7],[7,8,9,9]])

for n in range(0,3):
    znew = z[n,:]-z[n,0]

znew应为np.array([[0,1,2,3],[0,1,2,3],[0,1,2,2]])。我怎样才能做到这一点？这看起来微不足道。

Answer 1

您可以利用广播来实现这一目标：

>>> z - z[:,0][:, None] # or z - z[:,0][:, np.newaxis]
array([[0, 1, 2, 3],
       [0, 1, 2, 3],
       [0, 1, 2, 2]])