numpy.extract

原文：https://docs.scipy.org/doc/numpy/reference/generated/numpy.extract.html

译者：飞龙 UsyiyiCN

校对：（虚位以待）

numpy.extract(condition, arr)[source]

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

这相当于np.compress（ravel（condition）， ravel（arr））。如果条件是布尔np.extract等效于arr[condition]。

请注意，place与extract完全相反。

参数：	condition：array_like 数组，其非零或True条目表示要提取的arr的元素。 arr：array_like 输入与条件相同大小的数组。
返回：	提取：ndarray 来自arr的值1的数值组，其中条件为真。

也可以看看

take, put, copyto, compress, place

例子

>>> 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]], dtype=bool)
>>> np.extract(condition, arr)
array([0, 3, 6, 9])

如果条件是布尔值：

>>> arr[condition]
array([0, 3, 6, 9])