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

numpy.ma.choose()函数使用索引数组从一组选择中构造一个新数组。给定一个整数数组和一组n个选择数组，此方法将创建一个合并每个选择数组的新数组。如果arr中的arr值是i，则新数组将具有choices [i]包含在同一位置的值。

用法： numpy.ma.choose(arr, choices, out = None, mode = ‘raise’)

参数：
arr:[int的整数数组]此数组必须包含[0，n-1]中的整数，其中n是选择的数目。
choices:[数组的顺序]选择数组。索引数组和所有选择都应广播为相同形状。
out:[数组，可选]如果提供，结果将插入到此数组中。它应该具有适当的形状和dtype。
mode:[{‘raise’，‘wrap’，‘clip’}，可选]指定out-of-bounds索引的行为。 ‘raise’：引发错误。 ‘wrap’：环绕。 ‘clip’：剪切到范围。

Return :一个合并每个选择数组的新数组。

代码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]

相关用法

注：本文由纯净天空筛选整理自sanjoy_62大神的英文原创作品 numpy.ma.choose() function – Python。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。