sorted函数

LITElric 2019-07-01

一、需要用到的函数

需要用到的函数很少,基本上就是sorted()和lambda。

sorted()函数原型:

sorted(iterable[,key][,reverse])

可见sorted()可以接受3个参数,需要排序的变量必不可少,然后是key指定排序的元素,reverse指定是否逆序。

key:一个只有一个参数的函数,这个函数会被用在序列里的每一个元素上,所产生的结果将是排序算法依赖的对比关键字。比如说,在对一些字符串排序时,可以用key=len来实现根据字符串的长度进行排序。这个参数的默认值是恒等函数,也就是默认用元素自己的值来排序 (key=lambda x: x)。

lambda实际上就是匿名函数

一般形式:

lambda arguments: expression

写成函数形式就是

def <lambda>(arguments):
    return expression

当lambda和sorted()的key参数结合就有意想不到的效果了。

二、各种排序

1.简单list排序

lis = ['a', 'b', 'c']
print(sorted(lis))
# ['a', 'b', 'c']
print(sorted(lis, reverse=True))
# ['c', 'b', 'a']

2.dict的key排序

dic = {'c': 1, 'b': 2, 'a': 3} #字典排序默认按照字典的key排序,返回的也是字典的key
print(sorted(dic))
# ['a', 'b', 'c']
print(sorted(dic, reverse=True))
# ['c', 'b', 'a']

3.dict的value排序

dic = {'c': 1, 'b': 2, 'a': 3}
print(sorted(dic, key=lambda k: dic[k])) #dic([k])就是value
# ['c', 'b', 'a']
print(sorted(dic, key=lambda k: dic[k], reverse=True))
# ['a', 'b', 'c']

4.list内嵌套list排序

lis = [[4, 2, 9], [1, 5, 6], [7, 8, 3]]
print(sorted(lis, key=lambda k: k[0])) #列表内的没项代入key后的表达式,根据返回值排序
# [[1, 5, 6], [4, 2, 9], [7, 8, 3]]
print(sorted(lis, key=lambda k: k[1]))
# [[4, 2, 9], [1, 5, 6], [7, 8, 3]]
print(sorted(lis, key=lambda k: k[2]))
# [[7, 8, 3], [1, 5, 6], [4, 2, 9]]
print(sorted(lis, key=lambda k: k[0], reverse=True))
# [[7, 8, 3], [4, 2, 9], [1, 5, 6]]

5.dict内嵌套dict排序

dic = {
    'a': {'x': 3, 'y': 2, 'z': 1},
    'b': {'x': 2, 'y': 1, 'z': 3},
    'c': {'x': 1, 'y': 3, 'z': 2},
}
print(sorted(dic, key=lambda k: dic[k]['x']))
# ['c', 'b', 'a']
print(sorted(dic, key=lambda k: dic[k]['y']))
# ['b', 'a', 'c']
print(sorted(dic, key=lambda k: dic[k]['z']))
# ['a', 'c', 'b']
print(sorted(dic, key=lambda k: dic[k]['x'], reverse=True))
# ['a', 'b', 'c']

6.list内嵌套dict排序

lis = [
    {'x': 3, 'y': 2, 'z': 1},
    {'x': 2, 'y': 1, 'z': 3},
    {'x': 1, 'y': 3, 'z': 2},
]
print(sorted(lis, key=lambda k: k['x']))
# [{'z': 2, 'x': 1, 'y': 3}, {'z': 3, 'x': 2, 'y': 1}, {'z': 1, 'x': 3, 'y': 2}]
print(sorted(lis, key=lambda k: k['y']))
# [{'z': 3, 'x': 2, 'y': 1}, {'z': 1, 'x': 3, 'y': 2}, {'z': 2, 'x': 1, 'y': 3}]
print(sorted(lis, key=lambda k: k['z']))
# [{'z': 1, 'x': 3, 'y': 2}, {'z': 2, 'x': 1, 'y': 3}, {'z': 3, 'x': 2, 'y': 1}]
print(sorted(lis, key=lambda k: k['x'], reverse=True))
# [{'z': 1, 'x': 3, 'y': 2}, {'z': 3, 'x': 2, 'y': 1}, {'z': 2, 'x': 1, 'y': 3}]

7.dict内嵌套list排序

dic = {
    'a': [1, 2, 3],
    'b': [2, 1, 3],
    'c': [3, 1, 2],
}
print(sorted(dic, key=lambda k: dic[k][0]))
# ['a', 'b', 'c']
print(sorted(dic, key=lambda k: dic[k][1]))
# ['b', 'c', 'a']
print(sorted(dic, key=lambda k: dic[k][2]))
# ['c', 'b', 'a']
print(sorted(dic, key=lambda k: dic[k][0], reverse=True))
# ['c', 'b', 'a']

实例

1、给定一个只包含大小写字母,数字的字符串,对其进行排序,保证:

所有的小写字母在大写字母前面;
所有的字母在数字前面;
所有的奇数在偶数前面;

>>> s = "Sorting1234"
>>> "".join(sorted(s, key=lambda x: (x.isdigit(), x.isdigit() and int(x) % 2 == 0, x.isupper(), x.islower(), x)))
'ginortS1324'

2、排序一副FranchDeck。流畅的python书中示例

import collections

Card = collections.namedtuple('Card', ['rank', 'suit']) #命名元组

class FrenchDeck:
    ranks = [str(n) for n in range(2, 11)] + list('JQKA')  #2到10和JQKA
    suits = 'spades diamonds clubs hearts'.split()  #split默认按空格生成列表

    def __init__(self):
        self._cards = [Card(rank, suit) for suit in self.suits for rank in self.ranks] #生成所有52张牌

    def __len__(self):
        return len(self._cards)  #可以用Len()方法

    def __getitem__(self, position):
        return self._cards[position]  #可以用索引或切片

suit_values = dict(spades=3, hearts=2, diamonds=1, clubs=0) #设置花色优先级

def spades_high(card):
        rank_value = FrenchDeck.ranks.index(card.rank)  #找到当前Card所在的索引:2的索引为0
        return rank_value * len(suit_values) + suit_values[card.suit] #返回当前牌对应的优先级数字,如果没有乘于len(suit_values)那么('2', 'spades')和('4', 'diamonds')的优先级数字就一样了,这里也可以直接乘于4或5,6....
    
#排序
>>> import Frenchcard
>>> deck = Frenchcard.FrenchDeck()
>>> for i in sorted(deck, key=b.spades_high): #sorted返回一个列表。key后的函数会代入deck列表里的每一项返回一个优先级数字,并以这个优先级数字排序
...     print(i)
Card(rank='2', suit='clubs')
Card(rank='2', suit='diamonds')
Card(rank='2', suit='hearts')
Card(rank='2', suit='spades')
Card(rank='3', suit='clubs')
Card(rank='3', suit='diamonds')
.............

相关推荐