numpy：如何便宜地在numpy矩陣切片上應用布爾索引數組

Question

我正在嘗試在我的numpy矩陣的一部分上應用布爾掩碼數組。 我只是想出以下方法來實現此目的，首先將此切片復制到單個數組中，然后在此臨時數組上應用布爾掩碼，然后將其復制回到矩陣中。

我想知道是否存在（肯定存在！但是我找不到它）一種更便宜的實現方法？ 又名，無需復制成本。

#(data is a numpy 2d matrix)

tmp = data[i,:]
tmp[tmp==0] = mean
data[i,:] = tmp

Answer 1

那這個呢？

mask = data[i,:] == 0
data[i,mask] = mean