Python numpy.select()函数

numpy.select()()函数根据条件，从choicelist中的元素中返回一个数组。

语法: numpy.select(condlist, choicelist, default = 0)
参数 :
condlist : [bool ndarrays的列表] 它决定从choicelist中的哪个数组中提取输出元素。当满足多个条件时，使用在condlist中遇到的第一个条件。
choicelist : [list of ndarrays] 输出元素的数组列表。它的长度必须与condlist相同。
default : [标量，可选] 当所有条件评估为False时插入输出的元素。
返回： [ndarray] 根据条件，从choicelist中的元素中抽取一个数组。

代码#1：

# Python program explaining
# numpy.select() function
  
# importing numpy as geek 
import numpy as geek
  
arr = geek.arange(8)
  
condlist = [arr<3, arr>4]
choicelist = [arr, arr**3]
  
gfg = geek.select(condlist, choicelist)
  
print (gfg)

输出 :

[  0, 1, 2, 0, 0, 125, 216, 343]

代码#2：

# Python program explaining
# numpy.select() function
  
# importing numpy as geek 
import numpy as geek
  
arr = geek.arange(8)
  
condlist = [arr<4, arr>6]
choicelist = [arr, arr**2]
  
gfg = geek.select(condlist, choicelist)
  
print (gfg)