在numpy数组中快速就地替换某些值

Question

必须有更快的方法来替换值，对吧？我有一个2D数组代表一个高程/水深测量网格。我想用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

请参阅NumPy教程中的Indexing with Boolean Arrays。

Answer 2

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