find Nan values in an array code example

Example 1: check nan values in a np array

array_has_nan = np.isnan(array_sum)

Example 2: numpy find rows containing nan

# in each case returns array of bool
np.isnan(a).any(axis=1) # rows where any value is nan
np.isnan(a).all(axis=1) # rows where all values are nan