NumPy 函数

NumPy extract() 函数返回满足某些条件的数组元素。如果条件是布尔值，则该函数相当于arr[condition]。

语法

numpy.extract(condition, arr)

参数

condition必填。 指定一个数组，其非零或 True 条目指示要提取的 arr 元素。
arr必填。 指定与条件大小相同的数组（array_like）。

返回值

从条件为 True 的 arr 中返回排名 1 的值数组。

示例:

在下面的示例中，extract() 函数用于根据给定条件从数组中提取元素。

import numpy as np

Arr = np.arange(12).reshape((3, 4))

#显示数组
print("原始数组：")
print(Arr)

#定义条件
condition = np.mod(Arr, 3) == 0
print("\n条件为:")
print(condition)

#在数组上应用条件
print("\n根据条件提取元素:")
print(np.extract(condition, Arr))

上述代码的输出将是:

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

条件为:
[[ True False False  True]
 [False False  True False]
 [False  True False False]]

根据条件提取元素:
[0 3 6 9]