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])
>>> x
array([ 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()在一维和二维上运行的结果