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])