python排序函数

python 基础的排序函数 sort 、sorted 、argsort

一、sort 、sorted

1.简单升序排序直接调用sorted函数即可

sort()方法仅定义在list中

ls = list([5, 2, 3, 1, 4])
ls.sort()
print ls
>>>[1, 2, 3, 4, 5]

sorted()方法对所有的可迭代序列都有效,返回list

l1 = ['b','c','d','b','c','a','a'] 
sorted(set(l1),key=l1.index) 
Out[31]: ['b', 'c', 'd', 'a']

print(sorted({8: 'D', 2: 'B', 3: 'B', 4: 'E', 5: 'A'}))
>>>[2, 3, 4, 5, 8]

2.sorted(iterable，cmp，key，reverse）

(1) 使用cmp函数排序，cmp是带两个参数的比较函数

>>>L = [('david', 90), ('mary',90), ('sara',80),('lily',95)]

>>>print(sorted(L,cmp = lambda x,y: >>>cmp(x[0],y[0])))#按照第一个位置的字母序排序
[('david', 90), ('lily', 95), ('mary', 90), ('sara', 80)]

>>>print(sorted(L,cmp = lambda x,y: >>>cmp(x[1],y[1])))#按照第二个位置的数字序排序
[('sara', 80), ('david', 90), ('mary', 90), ('lily', 95)]

(2) 添加key参数，key 是带一个参数的函数
list.sort()和sorted()函数使用key参数来指定一个函数，此函数将在每个元素比较前被调用。

# 通过key指定的函数来忽略字符串的大小写
>>>print(sorted("This is a test string from Andrew".split(), key=str.lower))
['a', 'Andrew', 'from', 'is', 'string', 'test', 'This']

# 通过key指定的函数来保持原序列顺序
>>>l1 = ['b','c','d','b','c','a','a'] 
>>>sorted(set(l1),key=l1.index) 
Out[31]: ['b', 'c', 'd', 'a']

（3） reverse

>>>L = [('david', 90), ('mary',90), ('sara',80),('lily',95)]
>>>print(sorted(L, reverse=True))  # 逆转
[('sara', 80), ('mary', 90), ('lily', 95), ('david', 90)]

二、argsort

argsort 是 numpy 库中的函数。返回数组值从小到大的索引值。

# 一维数组
>>>x = np.array([5, 4, 2])
>>>np.argsort(x)
Out[45]: array([2, 1, 0])

# 二维数组
>>>x = np.array([[0, 3], [2, 2]])
array([[0, 3],
       [2, 2]])
>>>np.argsort(x, axis=0) #按列排序
Out[48]: 
array([[0, 1],
       [1, 0]])
>>>np.argsort(x, axis=1) #按行排序
Out[51]: 
array([[0, 1],
       [0, 1]])

>>> x = np.array([3, 1, 2])
>>> np.argsort(x) #按升序排列
array([1, 2, 0])
>>> np.argsort(-x) #按降序排列
array([0, 2, 1])
>>> x[np.argsort(x)] #通过索引值排序后的数组
array([1, 2, 3])
>>> x[np.argsort(-x)]
array([3, 2, 1])