📅  最后修改于: 2023-12-03 15:04:35.049000             🧑  作者: Mango
在Python中,使用numpy库可以进行各种数组操作,其中nanargmax函数可以返回数组中nan以外元素的最大值索引。下面我们来具体了解一下这个函数。
numpy.nanargmax(a, axis=None)
参数说明:
返回值:数组中nan以外元素的最大值索引
我们来看一个完整的实例。首先导入numpy库:
import numpy as np
定义一个包含nan的数组:
arr = np.array([1, 2, np.nan, 4, 5])
使用nanargmax函数获取最大值的索引:
result = np.nanargmax(arr)
print(result)
输出结果为:
4
我们也可以指定轴向,对于二维数组,我们可以分别计算每行和每列的数组元素的最大值索引。
arr = np.array([[1, 2, np.nan, 4, 5],
[6, 7, 8, np.nan, 10]])
row_max_index = np.nanargmax(arr, axis=1)
print(row_max_index)
输出结果为:
[4 4]
这里axis=1代表我们计算每行的最大值索引。同理,如果要计算每列的最大值索引,我们可以设置axis=0
。