【Python】 对数组或列表排序，求最大的N个数或最小的N个数

函数：

numpy.argsort()

输入：list, array, ndarray,mat

输出：输出排序后的索引（升序，不改变输入）

numpy.sort()

输入:list,array.ndarray,mat

输出：输出排序后的值（升序，不改变输入）

关于此函数的详细用法，请看numpy.argsort()，numpy.sort()

import numpy as np

#输入可以是list,array,ndarray

#list
a = [1,4,0,3,2]
b = np.argsort(a)
c = np.sort(a)                               #其余情况类似,不再列举
print(b)
print(c)

#array
a = np.array([1,4,0,3,2])
b = np.argsort(a)
print(b)

#ndarray
a = np.array([[1,3,2], [3,2,5], [5,4,1]])
b = np.argsort(a, axis = 1)                 #axis默认为-1,即最后一个维度
print(b)

#mat
a = np.mat([1,4,0,3,2])
b = np.argsort(a)
print(b)

输出：
#list
[2,0,4,3,1]
[0,1,2,3,4]
#array
[2,0,4,3,1]
#ndarray
[[0 2 1]
 [1 0 2]
 [2 1 0]]
#mat
[[2,0,4,3,1]]