numpy.extract¶
-
numpy.
extract
(condition, arr)[source]¶ 返回满足某些条件的数组的元素。
这相当于
np.compress(ravel(condition), ravel(arr))
。如果条件是布尔np.extract
等效于arr[condition]
。参数: condition:array_like
数组,其非零或True条目表示要提取的arr的元素。
arr:array_like
输入与条件相同大小的数组。
返回: 提取:ndarray
来自arr的值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]], dtype=bool) >>> np.extract(condition, arr) array([0, 3, 6, 9])
如果条件是布尔值:
>>> arr[condition] array([0, 3, 6, 9])