Python numpy.flatnonzero函数方法的使用

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

numpy.flatnonzero

numpy.flatnonzero(a)     [source]

在a的展平版本中返回非零的索引。

这等效于np.nonzero(np.ravel(a))[0]。

参数 :

a :array_like

输入数据。

返回值 :

res :ndarray

输出数组,包含非零的a.ravel()元素的索引。

例子

>>> x = np.arange(-2, 3)
>>> x
array([-2, -1,  0,  1,  2])
>>> np.flatnonzero(x)
array([0, 1, 3, 4])

使用非零元素的索引作为索引数组来提取这些元素:

>>> x.ravel()[np.flatnonzero(x)]
array([-2, -1,  1,  2])
推荐阅读
cjavapy编程之路首页