Python numpy.compress()

Python numpy.compress()函数返回一个数组的选定切片，沿着上述axis，满足一个axis。

语法:

numpy.compress(condition, array, axis = None, out = None)

参数 :

condition : [array_like]用户提取元素所依据的条件。
在input_array上应用condition，如果我们打印condition，它将返回一个arra
充满了真或假。数组元素是由具有
真值。
array : 输入数组。用户在input_array元素上应用条件
axis : [optional, int]表示要选择哪一个片断。
默认情况下，对扁平化的数组[1-D]工作。
out : [optional, ndarray]带有input_array元素的output_array。
满足的条件

返回 :

含有input_array元素的数组的副本。满足条件并沿着给定的轴线

# Python Program illustrating
# numpy.compress method
  
import numpy as geek
  
array = geek.arange(10).reshape(5, 2)
print("Original array : \n", array)
  
a = geek.compress([0, 1], array, axis=0)
print("\nSliced array : \n", a)
  
a = geek.compress([False, True], array, axis=0)
print("\nSliced array : \n", a)

输出 :

Original array : 
 [[0 1]
 [2 3]
 [4 5]
 [6 7]
 [8 9]]

Sliced array : 
 [[2 3]]

Sliced array : 
 [[2 3]]