在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教程中的使用布尔数组建立索引 。

Answer 2

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