numpy.ma.where

numpy.ma.where(condition, x=<class numpy._globals._NoValue at 0x40b6a26c>, y=<class numpy._globals._NoValue at 0x40b6a26c>) [source]

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

Возвращает массив с элементами, сформированный по форме условия, где элементы берутся из x когда condition истинно, и из y в противном случае. Если ни x ни y не заданы, функция возвращает кортеж индексов, где condition истинно (результат condition.nonzero()).

Параметры:	condition : array_like, bool Условие, которое должно быть выполнено. Для каждого истинного элемента возвращается соответствующий элемент из x, в противном случае из y. x, y : array_like, необязательно Значения для выбора. x и y должны иметь такую же форму, как условие, или быть совместимыми для трансляции к этой форме.
Возвращает:	out : MaskedArray или кортеж ndarrays Результат — массив со скрытыми элементами, если x и y были заданы, иначе — результат condition.nonzero().

См. также

numpy.where

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

Примеры

>>> 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]]
>>> np.ma.where(x > 5)    # return the indices where x > 5
(array([2, 2]), array([0, 2]))

>>> print(np.ma.where(x > 5, x, -3.1416))
[[-3.1416 -- -3.1416]
 [-- -3.1416 --]
 [6.0 -- 8.0]]