📅  最后修改于: 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]]