Python numpy.ma.choose()函数

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]

Python教程

Java教程

Web教程

数据库教程

图形图像教程

大数据教程

开发工具教程

计算机教程