extract函数python_Python numpy.extract函数方法的使用

numpy.extract

numpy.extract(condition, arr)     [source]

返回满足某些条件的数组元素。

这等效于np.compress(ravel(condition)，ravel(arr))。 如果条件为布尔值，则np.extract等同于arr[condition]。

注意：place和extract正好相反。参数 ：condition ：array_like

一个数组，其非零项或True项表示要extract的arr元素。

arr ：array_like

与条件大小相同的输入数组。

返回值 ：extract ：ndarray

条件为True时，对数组中的1个值数组进行排名。

例子>>> arr = np.arange(12).reshape((3, 4))

>>> arr

array([[ 0, 1, 2, 3],

[ 4, 5, 6, 7],

[ 8, 9, 10, 11]])

>>> condition = np.mod(arr, 3)==0

>>> condition

array([[ True, False, False, True],

[False, False, True, False],

[False, True, False, False]])

>>> np.extract(condition, arr)

array([0, 3, 6, 9])

如果条件为布尔值：>>> arr[condition]

array([0, 3, 6, 9])