python中argsort，sort 和 sorted，operator.itemgetter函数

下面对argsort，sort 和 sorted这几个排序函数做一个简单的介绍

argsort（）

• 一维数组

argsort()函数，是numpy库中的函数
argsort()函数是将数组中的元素从小到大排列，提取其对应的index(索引)，然后输出：

>>> from numpy import *>>> x=array([1,4,3,-1,6,9])>>> xarray([ 1,  4,  3, -1,  6,  9])>>> argsort(x) # 升序排列array([3, 0, 2, 1, 4, 5], dtype=int64)>>> argsort(-x) # 按降序排列array([5, 4, 1, 2, 0, 3], dtype=int64)>>> x[argsort(x)] # 输出按索引值排序后的数组array([-1,  1,  3,  4,  6,  9])>>> x[argsort(-x)]array([ 9,  6,  4,  3,  1, -1])>>> 

结果表明输出是一维数组的元素按从小到大排列后，输出其在原数组中的索引值

• 二维数组

>>> array([[0, 3,4], [4,2, 2]])array([[0, 3, 4],       [4, 2, 2]])>>> argsort(x,axis=0) #按列排序array([[0, 1, 1],       [1, 0, 0]], dtype=int64)>>> argsort(x,axis=1) #按行排序array([[0, 1, 2],       [1, 2, 0]], dtype=int64)>>> 

由运行结果可以看出，对于二维数组的argsort（）排序，分别是按列和按行排序的，是针对单独的每一列和每一行的排序。

sorted（）和sort（）

python的内建排序函数有 sort、sorted两个。

基础的序列升序排序直接调用sorted()方法即可
需要注意：sort()方法仅定义在list中，而sorted()方法对所有的可迭代序列都有效，并且都是返回一个list

ls = [5, 2, 3, 1, 4]ls_2=list([5, 2, 3, 1, 4])print(sorted(ls))ls_2.sort()print(ls_2)print(sorted({8: 'D', 2: 'B', 3: 'B', 4: 'E', 5: 'A'}))

结果:

[1, 2, 3, 4, 5][1, 2, 3, 4, 5][2, 3, 4, 5, 8]

其中字典是按照key值排序

sorted(iterable，cmp，key，reverse）

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

参数：iterable可以是list或者iterator；
cmp是带两个参数的比较函数；
key 是带一个参数的函数；
reverse为False或者True；

（1）用cmp函数排序

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

结果：

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

（2）用key函数排序

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

>>> sorted(list1,key = lambda list1: list1[0]) # 按照第一个位置的字母序排序[('david', 90), ('lily', 95), ('mary', 90), ('sara', 80)]>>> sorted(list1,key = lambda list1: list1[1]) # 按照第二个位置的数字序排序[('sara', 80), ('david', 90), ('mary', 90), ('lily', 95)]>>> 

还可以通过key指定的函数来忽略字符串的大小写：

>>> sorted("This is a test string from Andrew".split(), key=str.lower)['a', 'Andrew', 'from', 'is', 'string', 'test', 'This']>>> 

（3）用reverse排序

通过参数reverse控制升序/降序

>>> a=[1,3,5,2,9,4,7,8,6,0]>>> a.sort(reverse=False)>>> a[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]>>> a.sort(reverse=True)>>> a[9, 8, 7, 6, 5, 4, 3, 2, 1, 0]>>> a=[1,3,5,2,9,4,7,8,6,0]>>> sorted(a,reverse=False)[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]>>> sorted(a,reverse=True)[9, 8, 7, 6, 5, 4, 3, 2, 1, 0]>>> a[1, 3, 5, 2, 9, 4, 7, 8, 6, 0]# 字母排序>>> list1 = [('david', 90), ('mary',90), ('sara',80),('lily',95)]>>> sorted(list1,reverse = True)[('sara', 80), ('mary', 90), ('lily', 95), ('david', 90)]>>> sorted(list1,reverse = False)[('david', 90), ('lily', 95), ('mary', 90), ('sara', 80)]>>> 

operator.itemgetter函数

operator模块提供的itemgetter函数用于获取对象的哪些维的数据，参数为一些序号（即需要获取的数据在对象中的序号）

m=[1,2,4]b=operator.itemgetter(1) //定义函数b，获取对象的第1个域的值print(b(m))b=operator.itemgetter(1,0)print(b(m))m=array([[0, 3,4], [4,2, 2]])b=operator.itemgetter(1) #定义函数b，获取对象的第1维的值print(b(m))     b=operator.itemgetter(1,0) #获取对象的第1维和第0维的值print(b(m))# 排序m={'A': 1, 'B': 2}mm=list(m.iteritems())print(mm)# 对m的value值进行倒序排列print(sorted(m.iteritems(), \            key=operator.itemgetter(1), reverse=True))b=operator.itemgetter(1)print(b(mm)) # 必须操作list

结果：

2(2, 1)[4 2 2](array([4, 2, 2]), array([0, 3, 4]))[('A', 1), ('B', 2)][('B', 2), ('A', 1)]  ('B', 2)

由此可以清楚地看出operator.itemgetter()在一维和二维上运行的结果