ma.
choose
使用索引数组从一组选项中构造新数组。
给定一个整数数组和一组n个选择数组,此方法将创建一个合并每个选择数组的新数组。其中一个值 a 是i,新数组将具有选择的值 [i] 包含在同一位置。
此数组中必须包含整数 [0, n-1] ,其中n是选项数。
[0, n-1]
选择数组。索引数组和所有选择都应该可以广播到相同的形状。
如果提供,则结果将插入此数组。它应该是合适的形状和 dtype .
指定越界索引的行为方式。
“raise”:引发错误
“环绕”:环绕
“剪辑”:剪辑到范围
参见
等效函数
实例
>>> 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)