Skip to content

Latest commit

 

History

History
57 lines (38 loc) · 1.18 KB

np126_0993.md

File metadata and controls

57 lines (38 loc) · 1.18 KB

numpy.extract

原文:numpy.org/doc/1.26/reference/generated/numpy.extract.html

numpy.extract(condition, arr)

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

这相当于np.compress(ravel(condition), ravel(arr))。如果condition是布尔类型,np.extract相当于arr[condition]

请注意,placeextract完全相反。

参数:

conditionarray_like

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

arrarray_like

condition大小相同的输入数组。

返回:

提取ndarray

condition为 True 时,从arr中提取的值为condition为 True 的值的一维数组。

另请参阅

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

如果condition是布尔类型:

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