numpy.ma.choose

numpy.ma.choose(indices, choices, out=None, mode='raise') [source]

Используйте массив индексов для создания нового массива из набора вариантов.

При заданном массиве целых чисел и наборе массивов вариантов этот метод создаст новый массив, объединяющий каждый из массивов вариантов. Там, где значение в a равно i, новый массив будет содержать значение, которое choices[i] содержит в том же месте.

Параметры:

a : ndarray целых чисел Этот массив должен содержать целые числа в [0, n-1], где n — количество вариантов. choices : последовательность массивов Массивы вариантов. Массив индексов и все варианты должны быть совместимы для трансляции в один и тот же размер. out : массив, необязательно Если задано, результат будет вставлен в этот массив. Он должен иметь соответствующую форму и dtype. mode : {‘raise’, ‘wrap’, ‘clip’}, необязательно Указывает, как будут вести себя индексы, выходящие за пределы границ. ‘raise’ : генерировать ошибку ‘wrap’ : оборачивать ‘clip’ : обрезать до диапазона

Возвращает:

merged_array : массив

См. также

choose

эквивалентная функция

Примеры

>>> choice = np.array([[1,1,1], [2,2,2], [3,3,3]])
>>> a = np.array([2, 1, 0])
>>> np.ma.choose(a, choice)
masked_array(data = [3 2 1],
      mask = False,
      fill_value=999999)