sostituire zeri in matrice numpy con il valore mediano

Question

Ho una matrice NumPy simili:

foo_array = [38,26,14,55,31,0,15,8,0,0,0,18,40,27,3,19,0,49,29,21,5,38,29,17,16] 

voglio sostituire tutti gli zeri con il valore mediano l'intero array (dove i valori zero non sono essere inclusi nel calcolo della mediana)

Finora ho questo succede:

foo_array = [38,26,14,55,31,0,15,8,0,0,0,18,40,27,3,19,0,49,29,21,5,38,29,17,16] 
foo = np.array(foo_array) 
foo = np.sort(foo) 
print "foo sorted:",foo 
#foo sorted: [ 0 0 0 0 0 3 5 8 14 15 16 17 18 19 21 26 27 29 29 31 38 38 40 49 55] 
nonzero_values = foo[0::] > 0 
nz_values = foo[nonzero_values] 
print "nonzero_values?:",nz_values 
#nonzero_values?: [ 3 5 8 14 15 16 17 18 19 21 26 27 29 29 31 38 38 40 49 55] 
size = np.size(nz_values) 
middle = size/2 
print "median is:",nz_values[middle] 
#median is: 26 

esiste un modo intelligente per raggiungere questo obiettivo con la sintassi numpy?

Grazie

Answer 1

Questa soluzione sfrutta numpy.median:

import numpy as np 
foo_array = [38,26,14,55,31,0,15,8,0,0,0,18,40,27,3,19,0,49,29,21,5,38,29,17,16] 
foo = np.array(foo_array) 
# Compute the median of the non-zero elements 
m = np.median(foo[foo > 0]) 
# Assign the median to the zero elements 
foo[foo == 0] = m 

Solo una nota di cautela, la mediana per l'array (senza zeri) è 23,5, ma come scritto questo bastoni in 23.

Answer 2

foo2 = foo[:] 
foo2[foo2 == 0] = nz_values[middle] 

Invece di foo2, si può solo aggiornare foo se si desidera. La sintassi degli smart array di Numpy può combinare alcune righe del codice che hai creato. Ad esempio, invece di,

nonzero_values = foo[0::] > 0 
nz_values = foo[nonzero_values] 

Si può solo fare

nz_values = foo[foo > 0] 

È possibile trovare maggiori informazioni su "indicizzazione fantasia" nel documentation.