📜  Python中的numpy.compress

📅  最后修改于: 2020-06-10 00:58:01             🧑  作者: Mango

numpy.compress(condition, array, axis = None, out = None) : 沿着满足上述要求的轴返回数组的选定切片。

参数:

condition:[array_like]条件,根据该条件用户提取元素。
在input_array上应用条件,如果我们打印条件,它将返回一个arra
填入True或False。数组元素是从具有以下特征的索引中提取的:
真正的价值。
array:输入数组。用户对input_array元素应用条件
axis:[可选,int]指示要选择的切片。
默认情况下,对扁平数组[1-D]进行处理
out:[可选,ndarray]带有input_array元素的Output_array,
满足条件

 

返回:

带有input_array的元素的满足条件并沿给定轴的数组副本 

# Python程序说明numpy.compress方法 
  
import numpy as geek 
  
array = geek.arange(10).reshape(5, 2) 
print("原始数组 : \n", array) 
  
a = geek.compress([0, 1], array, axis=0) 
print("\n切片数组 : \n", a) 
  
a = geek.compress([False, True], array, axis=0) 
print("\n切片数组 : \n", a) 

输出: 

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

切片数组:
 [[2 3]] 

切片数组:
 [[2 3]]