Szybkie zastąpienie niektórych wartości w miejscu w tablicy numpy

Question

Musi istnieć szybszy sposób na zastąpienie wartości, prawda? Mam tablicę 2D reprezentującą siatkę rzędnych/batymetrii. Chcę wymienić niczego na 0 z NAN i w ten sposób jest bardzo powolny:

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

Wydaje się, że ma tak być o wiele lepszy sposób!

Answer 1

Następujące będzie to zrobić:

elevation[elevation > 0] = numpy.NAN 

Zobacz Indexing with Boolean Arrays w tutorialu NumPy.

Answer 2

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