NumPy参考 >例行程序 >Sorting, searching, and counting > numpy.extract
numpy.
extract
(condition,arr )[源代码] ¶返回满足某些条件的数组元素。
这等效于。如果
condition为boolean 等效于。np.compress(ravel(condition), ravel(arr))
np.extract
arr[condition]
一个非零或True条目指示 要提取的arr元素的数组。
输入数组的大小与condition相同。
等级1个数组值从ARR其中条件是真。
例子
>>> 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])