numpy.ma.masked_where

numpy.ma.masked_where(condition, a, copy=True)

Return a as an array masked where condition is True. Masked values of a or condition are kept.

Parameters:

condition : array_like

Masking condition.

a : array_like

Array to mask.

copy : bool

Whether to return a copy of a (True) or modify a in place (False).