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]

最后

以上就是大方白昼为你收集整理的numpy.argsort()函数numpy.argsort()函数的全部内容，希望文章能够帮你解决numpy.argsort()函数numpy.argsort()函数所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。