本文主要介绍了Python常用的内置数据结构的各种排序方式。
列表的排序
列表元素如下:
lst = [
[1, 10],
[1, 5],
[2, 10]
]
按照嵌套列表的第二个元素进行升序:
sorted(lst, key=lambda x: x[1])
# Output
[[1, 5], [1, 10], [2, 10]]
先按照嵌套列表的第一个元素进行升序,第一个元素相同的再按照第二个元素进行升序:
sorted(lst, key=lambda x: (x[0], x[1]))
# Output
[[1, 5], [1, 10], [2, 10]]
同理,我们也可以先按照第一个元素进行排序,然后第一个元素相同的再按照第二个元素进行排序,只需要把两个元素的位置颠倒即可:
sorted(lst, key=lambda x: (x[1], x[0]))
# Output
[[1, 5], [1, 10], [2, 10]]
我们都知道,如果想要降序(逆序)输出的时候,只需要带上reverse=True
参数即可,但是如果我们想要先按照嵌套列表的第一个元素进行升序,第一个元素相同的再按照第二个元素进行降序,两个元素的排序方式不同,这样应该怎么操作呢?
这其实也不难,因为第一个元素是升序,第二个元素是降序,同时元素都是数值,所以我们只需要对第二个排序元素进行取负值即可。因为一个数值取负数再进行升序,相当于原数值的降序。
sorted(lst, key=lambda x: (x[0], -x[1]))
# Output
[[1, 10], [1, 5], [2, 10]]
那如果两个元素都是字符串呢?这样就没办法像数值一样取负值了,还有办法实现上面类似的操作吗?
答案当然是可以的,不过这个时候我们还需要借助Python的一个内置函数ord
,它可以用于获取单个字符在Unicode中的位置,相当于我们把这个字符转成数值了,这样又可以通过取负值的方式进行逆序输出。
如果只有单个字符,那么可以这样实现即可:
# 元素是字符串
str_lst = [
['a', 'a'],
['a', 'b'],
['b', 'c']
]
# 先按照第一个元素升序,第一个元素相同的按照第二个元素字母降序
sorted(str_lst, key=lambda item: (item[0], -ord(item[1])))
# Output
[['a', 'b'], ['a', 'a'], ['b', 'c']]
如果是字符串,那么可以这样实现:
str_lst = [
['a', 'aa'],
['a', 'ab'],
['b', 'cd']
]
# 先按照第一个元素升序,第一个元素相同的按照第二个元素字母降序
sorted(str_lst, key=lambda item: (item[0], tuple(map(lambda x: -ord(x), item[1]))))
# Output
[['a', 'ab'], ['a', 'aa'], ['b', 'cd']]
上面虽然是针对列表进行介绍,但是元组的排序方式也是同理的。
字典的排序
上面讨论完了列表的排序,接下来介绍字典元素的排序。
因为Python的内置方法本质上是不支持字典的排序的,所以我们需要先把字典转换成序列,然后才能使用sorted
方法进行排序,排序完再重新组合成字典。
字典元素如下:
# 字典
d = {
'a': 1,
'b': 10,
'c': 10
}
按照字典键进行升序:
key_lst = sorted(d) # ['a', 'b', 'c']
{k: d[k] for k in key_lst}
# Output
{'a': 1, 'b': 10, 'c': 10}
按照字典值进行升序:
tmp_lst = sorted(d.items(), key=lambda e: e[1]) # [('a', 1), ('b', 10), ('c', 10)]
{k: v for k, v in tmp_lst}
# Output
{'a': 1, 'b': 10, 'c': 10}
先按照value降序,value相同的再按照key升序:
tmp_lst = sorted(d.items(), key=lambda e: (-e[1],e[0]))
{k: v for k, v in tmp_lst}
# Output
{'b': 10, 'c': 10, 'a': 1}
其他更加复杂的排序方式,可以参考前面介绍的列表的排序方式。
小结
本文内容小结:
- 如何按照指定元素进行排序?按照key或者value进行排序?
- 如何对不同元素采用不同的排序方式?