Python numpy.nanargmax函数方法的使用

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

numpy.nanargmax

numpy.nanargmax(a, axis=None)     [source]

返回指定轴上最大值的索引,忽略NaN。 对于所有NaN切片,会引发ValueError。 警告:如果切片仅包含NaN和-Infs,则结果不可信任。

参数 :

a :array_like

输入数据。

axisint, 可选

沿其运行的轴。 默认情况下,使用拼合输入。

返回值 :

index_arrayndarray

索引数组或单个索引值。

例子

>>> a = np.array([[np.nan, 4], [2, 3]])
>>> np.argmax(a)
0
>>> np.nanargmax(a)
1
>>> np.nanargmax(a, axis=0)
array([1, 0])
>>> np.nanargmax(a, axis=1)
array([1, 1])
推荐阅读
cjavapy编程之路首页