numpy.ma.choose

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

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

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

Параметры

indicesndarray целых чисел

Этот массив должен содержать целые числа в [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)