numpy. 提取#

麻木的。extract (条件, arr ) [来源] #

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

这相当于.如果 条件是布尔值则相当于.np.compress(ravel(condition), ravel(arr))np.extractarr[condition]

请注意, 的place作用与 完全相反extract

参数
类似条件数组

一个数组,其非零或 True 条目指示要提取的arr元素 。

arr类似数组

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

返回
提取ndarray

条件为 True时,将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]])
>>> np.extract(condition, arr)
array([0, 3, 6, 9])

如果条件是布尔值:

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