numpy.ma.where

numpy.ma.where(condition, x=<no value>, y=<no value>) [source]

Возвращает массив с элементами из x или y, в зависимости от условия.

Примечание

Когда предоставлен только condition, эта функция идентична nonzero. Остальная часть документации описывает только случай, когда указаны все три аргумента.

Параметры:	condition : array_like, bool Где True, возвращает x, в противном случае возвращает y. x, y : array_like, optional Значения для выбора. x, y и condition должны быть согласуемы с некоторой формой.
Возвращает:	out : MaskedArray Маскарованный массив с masked элементами, где условие маскировано, элементами из x где condition равно True и элементами из y в противном случае.

См. также

numpy.where

Эквивалентная функция в основном модуле NumPy.

nonzero

Функция, которая вызывается, когда x и y отсутствуют

Примеры

>>> x = np.ma.array(np.arange(9.).reshape(3, 3), mask=[[0, 1, 0],
...                                                    [1, 0, 1],
...                                                    [0, 1, 0]])
>>> print(x)
[[0.0 -- 2.0]
 [-- 4.0 --]
 [6.0 -- 8.0]]
>>> print(np.ma.where(x > 5, x, -3.1416))
[[-3.1416 -- -3.1416]
 [-- -3.1416 --]
 [6.0 -- 8.0]]