python - 快速就地替换 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 教程中。