isnan

[Andre Alexander Bell]

Hi Kenny,

On 10/12/2010 05:58 AM, Kenny wrote:
> I have an array A, and another one B with same dimention and size. Now I
> want to change the delete the elements in A where the same position in B
> is nan. How can I do.
> 
> Similar to the matlab code A(find(isnan(B)) = []

How about this:

>>> from numpy import NAN
>>> A = np.array([1,2,3,4,5])
>>> B = np.array([NAN,2,3,NAN,NAN])
>>> A = A[isnan(B)]
>>> A
array([1, 4, 5])

Regards


Andre