numpy.ma.choose#

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

使用索引数组从选择列表中构造一个新的数组。

给定一个整数数组和一个包含 n 个选择数组的列表,此方法将创建一个新的数组,该数组合并每个选择数组。在index中的值为 i 的位置,新数组将具有choices[i]在相同位置处包含的值。

参数:
indices整数型 ndarray

此数组必须包含 [0, n-1]范围内的整数,其中 n 是选择的数量。

choices数组序列

选择数组。索引数组和所有选择都应可广播到相同的形状。

out数组,可选

如果提供,结果将插入此数组中。它应该具有适当的形状和dtype

mode{'raise', 'wrap', 'clip'},可选

指定超出范围的索引的行为方式。

  • ‘raise’:引发错误

  • ‘wrap’:循环

  • ‘clip’:裁剪到范围内

返回值:
merged_array数组

另请参阅

choose

等效函数

示例

>>> import numpy as np
>>> 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)