原文链接:点击打开链接
[python学习] 语言基础—排序函数(sort()、sorted()、argsort()函数)
python的内建排序函数有 sort、sorted两个。
1、基础的序列升序排序直接调用sorted()方法即可
1 ls = list([5, 2, 3, 1, 4]) 2 new_ls = sorted(ls)
或者使用ls.sort()即可,直接将ls改变 3 print(new_ls)
需要注意:sort()方法仅定义在list中,而sorted()方法对所有的可迭代序列都有效
并且针对任何的可迭代序列,sorted()都是返回一个list,
1 print(sorted({8: 'D', 2: 'B', 3: 'B', 4: 'E', 5: 'A'})) 2 #输出为:[2, 3, 4, 5, 8]
sorted(iterable,cmp,key,reverse)
2、使用cmp函数排序,cmp是带两个参数的比较函数
1 list1 = [('david', 90), ('mary',90), ('sara',80),('lily',95)] 2 3 print(sorted(list1,cmp = lambda x,y: cmp(x[0],y[0])))#按照第一个位置的字母序排序 4 #[('david', 90), ('lily', 95), ('mary', 90), ('sara', 80)] 5 6 print(sorted(list1,cmp = lambda x,y: cmp(x[1],y[1])))#按照第二个位置的数字序排序 7 8 #[('sara', 80), ('david', 90), ('mary', 90), ('lily', 95)]
3、添加key参数,key 是带一个参数的函数
list.sort()和sorted()函数使用key参数来指定一个函数,此函数将在每个元素比较前被调用。
例如通过key指定的函数来忽略字符串的大小写
1 print(sorted("This is a test string from Andrew".split(), key=str.lower)) 2 #输出为:['a', 'Andrew', 'from', 'is', 'string', 'test', 'This']
4、用reverse排序
1 print(sorted(list1,reverse = True))#逆转 2 3 #[('sara', 80), ('mary', 90), ('lily', 95), ('david', 90)]
5、argsort()函数,是numpy库中的函数
下面的不想写了,直接复制过来留个档!
argsort函数
argsort函数返回的是数组值从小到大的索引值
Examples
--------
One dimensional array:一维数组
>>> x = np.array([3, 1, 2])
>>> np.argsort(x)
array([1, 2, 0])
Two-dimensional array:二维数组
>>> x = np.array([[0, 3], [2, 2]])
>>> x
array([[0, 3],
[2, 2]])
>>> np.argsort(x, axis=0) #按列排序
array([[0, 1],
[1, 0]])
>>> np.argsort(x, axis=1) #按行排序
array([[0, 1],
[0, 1]])
#######################################
例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])
另一种方式实现按降序排序:
>>> a = x[np.argsort(x)]
>>> a
array([1, 2, 3])
>>> a[::-1]
array([3, 2, 1])