快速就地更換一個numpy數組中的某些值

Question

需要更快的方法來替換值，對嗎？我有一個二維數組代表一個高程/水深網格。我想用NAN取代0以上的任何東西，這種方式超級慢：

for x in range(elevation.shape[0]): 
    for y in range(elevation.shape[1]): 
     if elevation[x,y] > 0: 
      elevation[x,y] = numpy.NAN 

看起來這樣有更好的方法！

Answer 1

下面將做到這一點：

elevation[elevation > 0] = numpy.NAN 

見Indexing with Boolean Arrays在NumPy的教程。

Answer 2

np.putmask(elevation, elevation > 0, np.nan)