首页 > 文章列表 > 列表排序:Python的sort、sorted和numpy.argsort方法详解

列表排序:Python的sort、sorted和numpy.argsort方法详解

Sorted sort argsort。
138 2023-06-10

在Python编程中,经常需要对列表或数组进行排序操作。Python提供了多种排序方法,包括sort、sorted和numpy.argsort等。本文将详细介绍这些排序方法的使用方法和注意事项。

一、sort方法
sort方法是Python列表中内置的方法,可以对列表进行原地排序(即有返回值但不产生新的排序对象),不需要额外的导入库。sort方法的参数有两个:key和reverse。key表示排序时使用的键,reverse表示是否进行倒序排序。例如:

my_list = [3, 1, 4, 1, 5, 9, 2, 6, 5, 3, 5]
my_list.sort()
print(my_list) # 输出 [1, 1, 2, 3, 3, 4, 5, 5, 5, 6, 9]

my_list.sort(reverse=True)
print(my_list) # 输出 [9, 6, 5, 5, 5, 4, 3, 3, 2, 1, 1]

my_list = ["apple", "banana", "cherry", "orange"]
my_list.sort(key=lambda x: len(x))
print(my_list) # 输出 ["apple", "cherry", "orange", "banana"]

以上例子中,第一个例子对整数列表进行排序,第二个例子进行倒序排序,第三个例子使用了lambda表达式对字符串列表按长度排序。

需要注意的是,sort方法是原地排序,即会改变原有列表的顺序,返回值是None,所以不能直接对排序后的列表执行操作,需要在排序前先创建一个副本或使用其他方法存储排序后的结果。

二、sorted函数
sorted函数是Python内置的函数,可以对列表、元组、字符串等排序,并且返回一个新的排序后的对象,不改变原始输入对象。sorted函数的参数与sort方法相同,包括key和reverse。例如:

my_list = [3, 1, 4, 1, 5, 9, 2, 6, 5, 3, 5]
new_list = sorted(my_list)
print(new_list) # 输出 [1, 1, 2, 3, 3, 4, 5, 5, 5, 6, 9]

new_list = sorted(my_list, reverse=True)
print(new_list) # 输出 [9, 6, 5, 5, 5, 4, 3, 3, 2, 1, 1]

my_list = ["apple", "banana", "cherry", "orange"]
new_list = sorted(my_list, key=lambda x: len(x))
print(new_list) # 输出 ["apple", "cherry", "orange", "banana"]

sorted函数的返回值可以是列表、元组、字符串等类型,根据输入类型决定返回结果的类型。

三、numpy.argsort方法
numpy.argsort方法是numpy中的方法,主要用来对numpy数组进行排序。argsort方法返回排序后的下标。numpy.argsort方法的参数也是key和reverse。例如:

import numpy as np

my_array = np.array([3, 1, 4, 1, 5, 9, 2, 6, 5, 3, 5])
sort_index = np.argsort(my_array)
print(sort_index) # 输出 [1 3 6 0 9 2 4 8 7 5 10]

sort_index = np.argsort(-my_array)
print(sort_index) # 输出 [5 7 4 2 0 3 6 8 9 1 10]

my_array = np.array(["apple", "banana", "cherry", "orange"])
sort_index = np.argsort([len(x) for x in my_array])
print(sort_index) # 输出 [0 2 3 1]

以上例子中,第一个例子对numpy数组进行升序排序,并返回排序后的下标。第二个例子进行降序排序,需要对数组取相反数。第三个例子对字符串数组按长度排序。

需要注意的是,numpy.argsort方法返回的是下标列表,需要通过下标来获取排序结果。

总结:
本文主要介绍了Python中的sort、sorted和numpy.argsort方法,这些方法可用于对Python中的列表和数组进行排序。其中sort方法和sorted函数可以对Python内置的对象进行排序,numpy.argsort方法则是numpy中的方法,主要用于对numpy数组进行排序。这些方法都可以使用key和reverse等参数来控制排序行为,应根据需要选用不同的排序方法。