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]])

Python subtract first element from each respective row of matrix