numpy take nans values code example

Example 1: np where nan

np.argwhere(np.isnan(x))

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