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