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