Come posso modificare in modo condizionale i valori in una matrice numpy prendendo in considerazione i numeri nan?

Question

La mia matrice è una matrice 2D ed è numpy.nan valori oltre i valori negativi e positivi:

>>> 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) 

e voglio sostituire tutti i numeri positivi con un numero e tutti i numeri negativi con un altro numero.

Come posso farlo usando python/numpy?

(Per la cronaca, la matrice è il risultato di geoimage, che voglio per effettuare una classificazione)

Answer 1

Il fatto che avete np.nan nella propria matrice non dovrebbe importare . Basta usare fantasia indicizzazione:

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

Se si desidera sostituire il vostro np.nans:

x[np.isnan(x)] = something_not_nan 

Maggiori informazioni sulla fantasia indicizzazione a tutorial e il NumPy documentation.

Answer 2

Prova:

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