Python numpy.ma.choose()函数
numpy.ma.choose()函数使用一个索引数组从一组选择中构建一个新数组。给定一个整数数组和一组n个选择数组,这个方法将创建一个新的数组,将每个选择数组合并。当arr中的值为i时,新的数组将在相同的地方有choice[i]包含的值。
语法: numpy.ma.choose(arr, choices, out = None, mode = ‘raise’)
参数 :
arr : [ndarray of ints] 这个数组必须包含[0, n-1]中的整数,其中n是选择的数量。
choices : [数组序列] 选择数组。索引数组和所有的选择应该可以广播到同一个形状。
out : [array, optional] 如果提供,结果将被插入这个数组中。它应该是适当的形状和dtype。
mode : [{‘raise’, ‘wrap’, ‘clip’}, optional] 指定界外指数的行为方式。’raise’: 提出一个错误。’wrap’: 绕开。’clip’: 夹到范围内。
返回:一个新的数组,它合并了每个选择数组。
代码#1:
# Python program explaining
# numpy.ma.choose() function
# importing numpy as geek
# and numpy.ma module as ma
import numpy as geek
import numpy.ma as ma
choice = geek.array([[1, 1, 1], [2, 2, 2], [3, 3, 3]])
arr = geek.array([2, 1, 0])
gfg = geek.ma.choose(arr, choice)
print (gfg)
输出 :
[3 2 1]
代码#2:
# Python program explaining
# numpy.ma.choose() function
# importing numpy as geek
# and numpy.ma module as ma
import numpy as geek
import numpy.ma as ma
choice = geek.array([[1, 1, 1], [2, 2, 2], [3, 3, 3]])
arr = geek.array([0, 1, 2])
gfg = geek.ma.choose(arr, choice)
print (gfg)
输出 :
[1 2 3]