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]

你可能感兴趣的:(numpy,python,开发语言)