fonctionnalité "isnotnan" en numpy, cela peut-il être plus pythonique?

Question

J'ai besoin d'une fonction qui renvoie des valeurs non-NaN d'un tableau. Actuellement, je fais de cette façon:

>>> a = np.array([np.nan, 1, 2]) 
>>> a 
array([ NaN, 1., 2.]) 

>>> np.invert(np.isnan(a)) 
array([False, True, True], dtype=bool) 

>>> a[np.invert(np.isnan(a))] 
array([ 1., 2.]) 

Python: 2.6.4 numpy: 1.3.0

S'il vous plaît partager si vous connaissez une meilleure façon, Merci

Answer 1

a = a[~np.isnan(a)] 

Answer 2

Vous sont en train de tester pour tout ce qui n'est pas NaN et mtrw a la bonne façon de le faire. Si vous êtes intéressé par les tests pour les nombres finis (n'est pas NaN et ne INF) alors vous n'avez pas besoin d'inversion et peut utiliser:

np.isfinite(a) 

Plus pythonique et natif, d'une lecture facile, et souvent lorsque vous voulez éviter NaN, vous voulez également éviter INF dans mon expérience.

Juste pensé que je jetterais là-bas pour les gens.