📜  Python中的numpy.extract

📅  最后修改于: 2020-06-09 04:50:14             🧑  作者: Mango

numpy.extract(condition, array) :  如果满足某些指定条件,则返回input_array的元素。

参数:

array:输入数组。用户对input_array元素应用条件
condition:[array_like]条件,根据该条件用户提取元素。在input_array上应用条件,如果我们打印条件,它将返回一个数组填入True或False。数组元素是从具有以下特征的索引中提取的:真正的价值。

返回:

满足条件的数组元素。

# Python程序说明numpy.compress方法 
  
import numpy as geek 
  
array = geek.arange(10).reshape(5, 2) 
print("原始数组 : \n", array) 
  
a = geek.mod(array, 4) !=0
# 这将显示满足条件的元素状态 
print("\n数组条件a : \n", a) 
  
# 这将返回满足条件“ a"的元素 
print("\n满足条件a的元素  : \n", geek.extract(a, array)) 
  
  
  
b = array - 4 == 1
# 这将显示满足条件的元素状态 
print("\n数组条件b : \n", b) 
  
# 这将返回满足条件“ b"的元素 
print("\n满足条件b的元素  : \n", geek.extract(b, array)) 

输出:

原始数组:
 [[0 1] 
 [2 3] 
 [4 5] 
 [6 7] 
 [8 9]] 

数组条件a:
 [[False True] 
 [True True] 
 [False True] 
 [True True] 
 [False True] ] 

满足条件a的元素:
 [1 2 3 5 6 7 9] 

数组条件b:
 [[False False] 
 [False False] 
 [False True] 
 [False False] 
 [False False]] 

满足条件b的元素:
 [5 ]