Jak mogę warunkowo zmienić wartości w tablicy numpy, biorąc pod uwagę liczby nan?

Question

Moja macierz jest macierzą 2D i to numpy.nan wartości oprócz wartości ujemnych i dodatnich:

>>> array 
array([[  nan,   nan,   nan, ..., -0.04891211, 
      nan,   nan], 
    [  nan,   nan,   nan, ...,   nan, 
      nan,   nan], 
    [  nan,   nan,   nan, ...,   nan, 
      nan,   nan], 
    ..., 
    [-0.02510989, -0.02520096, -0.02669156, ...,   nan, 
      nan,   nan], 
    [-0.02725595, -0.02715945, -0.0286231 , ...,   nan, 
      nan,   nan], 
    [  nan,   nan,   nan, ...,   nan, 
      nan,   nan]], dtype=float32) 

I chcę wymienić wszystkie pozytywne numery z numerem i wszystkie negatywne numery z innym numerem.

Jak mogę to zrobić używając Pythona/Numpy?

(Dla przypomnienia, matryca jest wynikiem geoimage, które chcę wykonać klasyfikację)

Answer 1

Fakt, że masz np.nan w swojej tablicy nie powinno znaczenia . Wystarczy użyć fantazyjne indeksowania:

x[x>0] = new_value_for_pos 
x[x<0] = new_value_for_neg 

Jeśli chcesz wymienić np.nans:

x[np.isnan(x)] = something_not_nan 

Więcej informacji na temat indeksowania fantazyjny a tutorial i NumPy documentation.

Answer 2

Spróbuj:

a[a>0] = 1 
a[a<0] = -1