Python numpy ma.choose用法及代码示例

本文简要介绍 python 语言中 numpy.ma.choose 的用法。

用法:

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

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

给定一个整数数组和一个包含 n 个选择数组的列表，此方法将创建一个新数组来合并每个选择数组。当索引中的值为 i 时，新数组将具有choices[i] 在同一位置包含的值。

参数：

indices： 整数数组

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

choices： 数组序列

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

out： 数组，可选

如果提供，结果将被插入到这个数组中。它应该具有适当的形状和 dtype 。

mode： {‘raise’, ‘wrap’, ‘clip’}，可选

指定越界索引的行为方式。

• ‘raise’：引发错误

• ‘wrap’：环绕

• ‘clip’：剪辑到范围

返回：

merged_array： 数组

例子：

>>> 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)