一次性把Python中的排序方法都学透!

这是「活用数据」的第 13 篇原创文章

本文主要介绍了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进行排序?
  • 如何对不同元素采用不同的排序方式?

本文完整示例代码及文件可关注公众号,回复“python”获取。

你可能感兴趣的:(Python,算法,python,排序算法)