>>> a = np.arange(3.0) >>> a array([ 0., 1., 2.]) >>> a[a==0] array([ 0.]) >>> a[a==0] = np.nan >>> a array([ nan, 1., 2.])