numpy.ma.masked_invalid

numpy.ma.masked_invalid(a, copy=True)

Mask the array for invalid values (NaNs or infs). Any preexisting mask is conserved.