Python中的numpy.extract

numpy.extract()是NumPy中的一个函数，可以用于提取数组内符合某些条件的元素，返回一个新的数组。

函数格式

numpy.extract(condition, arr)

参数说明

• condition：一个bool类型数组或者bool类型的表达式，用于判断数组内的元素是否符合条件。条件为True的元素将会被提取出来。
• arr：需要进行提取的数组。

示例

提取数组arr中所有大于等于3小于等于8的元素。

import numpy as np

arr = np.array([1,2,3,4,5,6,7,8,9])
condition = np.logical_and(arr>=3, arr<=8)
result = np.extract(condition, arr)

print(result)  # 输出：[3 4 5 6 7 8]

结论

numpy.extract()提供了一种便捷的方式从数组中提取符合特定条件的元素，可以用于数据筛选和处理。