介绍
在本文中,你将了解一些你以前可能不知道的关于排序的技巧。
目录
- Sorted() VS Sort() 用于列表、元组和字典
- 根据你的标准排序
- 对复杂对象的集合进行排序
- 用内置函数attrgetter进行排序
1. Sorted() VS Sort() 用于列表、元组和字典
在 Python 中,有两个主要函数sorted()和sort(),用于对列表、元组和字典等集合进行排序。为了理解它们之间的区别,让我们举一些例子。
- 排序列表
假设你有以下数字列表。
list_1 = [24, -54, -1, 4, 0, 76]
如果你想得到这个列表的排序版本,你可以使用sorted()如下函数:
list_1 = [[24, -54, -1, 4, 0, 76] sorted_list = sorted (list_1) print("old list is:", list_1) print("sorted list is:", sorted_list)
输出:
old list is: [24, -54, -1, 4, 0, 76] sorted list is: [-1,0,4,24,54,76]
在前面的示例中,你可以看到原始列表没有受到影响,而新的排序列表已使用新变量存储sorted_list。
如果要对原始列表进行排序,换句话说,在不需要新变量的情况下更改原始列表,你可以使用实例方法sort()。请参考以下示例。
list_1 = [24,54,-2,3,0,76] print("排序前:",list_1) li() print("排序后:",list_1)
输出:
排序前: [24,54,-2,3,0,76] 排序后: [-2,0,3,24,54,76]
sort()从前面的输出中,我们可以了解到方法和sorted()函数的主要区别是:
- sorted()函数返回一个新的排序列表,因此你可以将其分配给一个新变量。
- sort()方法对列表进行适当的排序,因此它不返回任何内容。
- 对元组进行排序
对于 Python 中的元组,只能使用sorted()函数,因为元组是不可变的数据类型。因此,sort()不是受支持的方法。
tup_1 = (24, 54, -1, 4, 0, 76) sorted_tuple = sorted(tup_1) print('原始元组是:', tup_1) print('排序结果是:', sorted_tuple)
输出:
原始元组为:(24, 54, -1, 4, 0, 76) 排序结果为:[-1, 0, 4, 24, 54, 76]
- 对字典进行排序
在字典的情况下,该sorted()函数将仅对字典键进行排序。让我们看一个简单的例子。
dic = {'course': 'Python Sorting', 'duration':'5 mins', 'trainer': 'Samer Sallam', 'level': 'Advanced'} sorted_dic = sorted(dic) print('排序结果:', sorted_dic)
输出:
排序结果:['course', 'duration', 'level', 'trainer']
请注意,排序的返回值是传递的字典键的排序列表。
无论你要排序什么,都可以使用参数"reverse = True"进行降序排序。
2.根据你的标准排序
在前面的示例中,项目是根据它们的实际值排序的,但是如果你想根据另一个标准对它们进行排序怎么办。例如,假设你想根据它们的绝对值对它们进行排序。
为此,你可以使用参数key传递代表你的标准的可调用函数。让我们看下一个使用内置函数的示例abs()。
list_1 = [24, -54, -1, 4, 0, -76] sorted_list = sorted(list_1, key=abs) print('原始列表为:', list_1) print('按绝对值排序的列表为:', sorted_list)
输出
原始列表为:[24, -54, -1, 4, 0, -76] 按绝对值排序的列表为: [0, -1, 4, 24, -54, -76]
现在项目已按升序排序,但根据它们的绝对值。
3.对象排序
前面的所有示例都涵盖了其中项目是数字的集合,但是如果项目是复杂对象怎么办。接下来,你将看到在这种情况下该怎么做。
假设你有以下Student类。此外,假设你有一个来自同一类的三个对象的列表,如下所示(__repr__已被覆盖以很好地打印该对象):
class Student: def __init__(self, name, age): = name = age def __repr__(self): return f'({}, {})' student1 = Student('John', 26) student2 = Student('Alex', 25) student3 = Student('Bob', 30) students_list = [student1, student2, student3]
如果我们尝试对students_list使用sorted()函数进行排序看看会发生什么
sorted_students = sorted(students_list)
输出:
TypeError: '<' not supported between instances of 'Student' and 'Student'
我们得到一个类型错误,因为解释器不知道如何对这些对象进行排序。
为了解决这个问题,我们应该向解释器解释如何对它们进行排序,这是通过key再次使用参数来完成的。此参数接受定义排序标准的函数。
在下面的示例中,假设我们要根据学生的姓名对对象进行排序(参见key_sort函数)。
class Student: def __init__(self, name, age): = name = age def __repr__(self): return f'({}, {})' student1 = Student('John', 26) student2 = Student('Alex', 25) student3 = Student('Bob', 30) students_list = [student1, student2, student3] # define the sorting key def key_sort(student): return # sort the students_list according to key_sort sorted_students = sorted(students_list, key=key_sort) print(sorted_students)
输出:
[(Alex, 25), (Bob, 30), (John, 26)]
此外,如果你熟悉 Python 中的 lambda 函数,则可以使用它来代替定义key_sort函数。请参考以下示例。
class Student: def __init__(self, name, age): = name = age def __repr__(self): return f'({}, {})' student1 = Student('John', 26) student2 = Student('Alex', 25) student3 = Student('Bob', 30) students_list = [student1, student2, student3] # using lambda function as the sorting key sorted_students = sorted(students_list, key= lambda student: ) print(sorted_students)
输出:
[(Alex, 25), (Bob, 30), (John, 26)]
4. 用内置函数attrgetter排序
在前面的示例中,我们定义了自己的函数来从对象中获取属性。我们也可以改用内置attrgetter函数。从它的名字可以理解为它获取的是required属性的值。你可以从operator模块中导入此功能。
现在,让我们看一个示例,其中将根据学生的年龄对项目进行排序。
from operator import attrgetter class Student: def __init__(self, name, age): = name = age def __repr__(self): return f'({}, {})' student1 = Student('John', 26) student2 = Student('Alex', 25) student3 = Student('Bob', 30) students_list = [student1, student2, student3] # sort the students_list according to key_sort sorted_students = sorted(students_list, key=attrgetter('age')) print(sorted_students)
输出:
[(Alex, 25), (Bob, 30), (John, 26)]
现在,让我们总结一下在本文中学到的知识。
- 在 Python中sorted()和sort()函数第一个返回一个新的排序集合,而第二个更改原始集合。
- 你可以使用该参数reverse按降序排序。
- 你可以使用该参数key来定义你自己的排序标准。
- attrgetter当你想将对象实例属性用作排序键时,这是一个有用的内置函数。