numpy.argsort()函数

numpy.argsort()函数

• numpy.argsort() 函数用于使用关键字kind指定的算法沿给定轴执行间接排序。 它返回一个与 arr 形状相同的索引数组，用于对数组进行排序，按升序排列

1.函数原型：

numpy.argsort(arr, axis=-1, kind=’quicksort’, order=None) 

2.参数

• arr:[array_like]，输入数组
• axis:[int or None]，排序的轴， 如果没有，数组在排序前被展平。 默认值为 -1，即沿最后一个轴排序。
• kind:[‘quicksort’, ‘mergesort’, ‘heapsort’]，选择算法， 默认为“快速排序”。
• order : [str or list of str] ,当 arr 是一个定义了字段的数组时，这个参数指定首先比较哪些字段，第二个等等。
• return: [index_array, ndarray] ,沿指定轴对 arr 排序的索引数组。如果 arr 是一维的，则 arr[index_array] 返回排序后的 arr。

示例1

import numpy as np
# input array
in_arr = np.array([2, -1, 9, 0, 6, 40])
print("Input unsorted array : ", in_arr)

out_arr = np.argsort(in_arr)
print("Output sorted array indices : ", out_arr)
print("Output sorted array : ", in_arr[out_arr])

• 输出

Input unsorted array :  [ 2 -1  9  0  6 40]
Output sorted array indices :  [1 3 0 4 2 5]
Output sorted array :  [-1  0  2  6  9 40]

示例2

import numpy as np

# input 2d array

in_arr = np.array([[2, 0, 10],[5, 4, 3]])
print("input array : ", in_arr)

# output sorted array indices
out_arr1 = np.argsort(in_arr, kind='mergesort', axis=0) # 列
print("Output sorted array indices along axis 0: ", out_arr1)
out_arr2 = np.argsort(in_arr, kind='heapsort', axis=1) # 行
print("Output sorteded array indices along axis 1: ", out_arr2)

• 输出

input array :  [[ 2  0 10]
 [ 5  4  3]]
Output sorted array indices along axis 0:  [[0 0 1]
 [1 1 0]]
Output sorteded array indices along axis 1:  [[1 0 2]
 [2 1 0]]

示例3

# get two largest value from numpy array
x=np.array([12,43,2,100,54,5,68])
print(x)
# using argsort get indices of value of arranged in ascending order
print(np.argsort(x))
#get two highest value index of array
print(np.argsort(x)[-2:])
# to arrange in ascending order of index
print(np.argsort(x)[-2:][::-1])
# to get highest 2 values from array
x[np.argsort(x)[-2:][::-1]]

• 输出

[ 12  43   2 100  54   5  68]
[2 5 0 1 4 6 3]
[6 3]
[3 6]
[100  68]