如何用NumPy得到一个数组的n个最大的值
让我们看看如何使用NumPy库获得一个数组的n个最大值的程序。为了从NumPy数组中获得最大的数值,我们首先要使用NumPy的numpy.argsort()函数对NumPy数组进行排序,然后应用负数索引的切分概念。
语法: numpy.argsort(arr, axis=-1, kind=’quicksort’, order=None)
返回: [index_array, ndarray] 沿着指定的轴对arr进行排序的索引数组。如果arr是一维的,那么arr[index_array]返回一个排序的arr。
让我们看一个例子。
示例1:从NumPy数组中获取第一大值。
输出:
例子2:从NumPy数组中获取3个最大的值。
输出: