numpy.argsort()函数
numpy.argsort()函数用于使用关键字kind指定的算法沿给定轴执行间接排序。 它返回一个与 arr 形状相同的索引数组，用于对数组进行排序，按升序排列
1.函数原型：
复制代码
1
2numpy.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
复制代码
1
2
3
4
5
6
7
8import 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])
- 输出
复制代码
1
2
3
4
5
6
7
8
9
10
11
12
13
14Input 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
复制代码
1
2
3
4
5
6
7
8
9
10import 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)
- 输出
复制代码
1
2
3
4
5
6
7
8
9
10
11
12
13input 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
复制代码
1
2
3
4
5
6
7
8
9
10
11
12# 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]]
- 输出
复制代码
1
2
3
4
5
6
7
8
9
10
11
12[ 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内容请搜索靠谱客的其他文章。
本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
发表评论 取消回复