numpy.ma.where#

ma.где(условие, x=<нет значение>, y=<нет значение>)[источник]#

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

Примечание

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

Параметры:
условиеarray_like, bool

Где True, выдать x, в противном случае выдает y.

x, yarray_like, необязательный

Значения, из которых выбирать. x, y и условие должны быть транслируемы к некоторой форме.

Возвращает:
выходMaskedArray

Маскированный массив с masked элементов, где условие замаскировано, элементы из x где условие равно True, и элементы из y в другом месте.

Смотрите также

numpy.where

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

nonzero

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

Примеры

>>> import numpy as np
>>> x = np.ma.array(np.arange(9.).reshape(3, 3), mask=[[0, 1, 0],
...                                                    [1, 0, 1],
...                                                    [0, 1, 0]])
>>> x
masked_array(
  data=[[0.0, --, 2.0],
        [--, 4.0, --],
        [6.0, --, 8.0]],
  mask=[[False,  True, False],
        [ True, False,  True],
        [False,  True, False]],
  fill_value=1e+20)
>>> np.ma.where(x > 5, x, -3.1416)
masked_array(
  data=[[-3.1416, --, -3.1416],
        [--, -3.1416, --],
        [6.0, --, 8.0]],
  mask=[[False,  True, False],
        [ True, False,  True],
        [False,  True, False]],
  fill_value=1e+20)