详解python排序的5种高级用法

来源:投稿 作者:Fairy

编辑:学姐

排序是编程中常用的操作之一。Python提供了多种排序方法,可以适用于不同的排序需求。

那么,今天我们将介绍Python中常用的 5 种列表排序方法

「1.使用sorted()函数和lambda表达式」

使用sorted()函数和lambda表达式进行自定义排序:可以使用sorted()函数和lambda表达式来根据需要对列表、元组或字典等数据结构进行自定义排序。

# 对元组列表按照第一个元素升序排序
my_list = [(2, 'b'), (3, 'c'), (1, 'a')]
sorted_list = sorted(my_list, key=lambda x: x[0])
print(sorted_list) # 输出 [(1, 'a'), (2, 'b'), (3, 'c')]

首先,使用 sorted() 函数对 my_list 进行排序,并将排序后的结果赋值给变量 sorted_list。在 sorted() 函数中,使用 key 参数对排序依据进行定义,这里使用 lambda 表达式创建了一个简单的匿名函数,该函数接受一个参数 x,并返回元组中的第一个元素 x[0]。因此,sorted() 函数根据元组的第一个元素进行排序。

最后,打印出排好序的列表 sorted_list,最终结果如下:

详解python排序的5种高级用法_第1张图片

「2.使用operator模块」

这里需要先将operator模块导入脚本中,可以使用其中的函数(如itemgetter(),attrgetter())对复杂的数据类型进行排序。

import operator

# 对字典按照值降序排序
my_dict = {'a': 4, 'b': 2, 'c': 7}
sorted_dict = dict(sorted(my_dict.items(), key=operator.itemgetter(1), reverse=True))
print(sorted_dict) # 输出 {'c': 7, 'a': 4, 'b': 2}

首先,使用 items() 方法将字典转换为元组列表,每个元组包含一个键和一个值。然后使用 sorted() 函数对该列表进行排序。key 参数指定排序依据,operator.itemgetter(1) 表示按照元组的第二个元素(即字典的值)进行排序,reverse=True 表示进行降序排序。排序后的结果是一个元组列表,每个元组包含排好序的键值对。

最后,使用 dict() 函数将排好序的元组列表转换回字典格式,并将结果赋值给变量 sorted_dict。打印出 sorted_dict,最终结果如下:

详解python排序的5种高级用法_第2张图片

「3.使用heapq模块」

heapq是一个Python内置模块,提供了对堆的基本支持。它可以对可迭代对象进行原地堆排序并返回一个排序后的列表。

import heapq

# 找出列表中前3个最小的数字
my_list = [5, 8, 3, 9, 1, 6, 4]
smallest_numbers = heapq.nsmallest(3, my_list)
print(smallest_numbers) # 输出 [1, 3, 4]

使用 heapq.nsmallest() 函数可以从一个可迭代对象中,快速地找到其中的最小元素,同时可以指定要返回的最小元素个数。该函数接受两个参数:第一个参数是要返回的元素个数(即本例中的 3),第二个参数是一个可迭代对象(即本例中的列表 my_list)。

在本例中,我们将列表 my_list 传递给 heapq.nsmallest() 函数,并请求找出前 3 个最小的数字。smallest_numbers 变量存储了找到的最小数字,结果为 [1, 3, 4]。这些数字是按照升序排列的,符合预期的结果,最终结果如下:

详解python排序的5种高级用法_第3张图片

「4.使用functools模块」

在某些情况下,可能需要使用多个条件对数据进行排序。可以使用functools模块中的cmp_to_key()函数来实现这一点。

import functools

# 对元组列表按照第一个元素升序排序,如果第一个元素相同则按照第二个元素降序排序
my_list = [(2, 6), (3, 8), (1, 5), (2, 3)]
sorted_list = sorted(my_list, key=functools.cmp_to_key(lambda x, y: x[0] - y[0] or y[1] - x[1]))
print(sorted_list)

lambda 函数接受两个参数 x 和 y,分别表示要比较的元组。首先,它通过比较两个元组的第一个元素进行排序,即 x[0] - y[0];如果第一个元素相同,则再按照第二个元素降序排列,即 y[1] - x[1]。因为默认情况下 sorted 函数是升序排列,所以在第二个比较时需要对参数位置进行反转。

最后,使用 sorted() 函数对 my_list 进行排序,并将结果存储在 sorted_list 变量中。打印输出 sorted_list ,最终结果如下:

详解python排序的5种高级用法_第4张图片

「5.使用numpy模块」

numpy.sort(array)将按升序对数组进行排序。

import numpy as np

# 对数组进行升序排序
my_array = np.array([5, 8, 3, 9, 1, 6, 4])
sorted_array = np.sort(my_array)
print(sorted_array) # 输出 [1 3 4 5 6 8 9]

我们先创建了一个一维数组 my_array,然后调用 np.sort() 函数对该数组进行升序排序。排序结果被存储在 sorted_array 数组中。最后使用 print() 函数打印输出 sorted_array ,最终结果如下:

详解python排序的5种高级用法_第5张图片

关注下方《学姐带你玩AI》

回复“python”了解更多学习资料

码字不易,欢迎大家点赞评论收藏!

你可能感兴趣的:(深度学习干货,粉丝的投稿,人工智能干货,python,算法,列表排序)