Python numpy.extract函数方法的使用

NumPy(Numerical Python的缩写)是一个开源的Python科学计算库。使用NumPy,就可以很自然地使用数组和矩阵。NumPy包含很多实用的数学函数,涵盖线性代数运算、傅里叶变换和随机数生成等功能。本文主要介绍一下NumPy中extract方法的使用。

numpy.extract

numpy.extract(condition, arr)     [source]

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

这等效于np.compress(ravel(condition),ravel(arr))。 如果条件为布尔值,则np.extract等同于arr[condition]

注意placeextract正好相反。

参数 :

condition :array_like

一个数组,其非零项或True项表示要extract的arr元素。

arr :array_like

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

返回值 :

extract :ndarray

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

如果条件为布尔值:

>>> arr[condition]
array([0, 3, 6, 9])
推荐阅读
cjavapy编程之路首页