Thay thế tại chỗ nhanh chóng của một số giá trị trong một mảng nhiều mảng

Question

Có phải có cách nhanh hơn để thực hiện thay thế các giá trị, đúng không? Tôi đã có một mảng 2D đại diện cho một mạng lưới độ cao/độ sâu. Tôi muốn thay thế bất kỳ thứ gì trên 0 bằng NAN và cách này là siêu chậm:

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

Có vẻ như đó là cách tốt hơn!

Answer 1

Sau đây sẽ làm điều đó:

elevation[elevation > 0] = numpy.NAN 

Xem Indexing with Boolean Arrays trong hướng dẫn NumPy.

Answer 2

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