在 Python 中对二维数组进行排序

1. 使用 Python 中的 sort() 函数按列号对二维数组进行排序
2. 使用 Python 中的 sorted() 函数按列号对二维数组进行排序

我们将介绍在 Python 中对多维数组进行排序的不同方法。

数组排序内置了 sort() 和 sorted() 等函数；这些函数还允许我们获取一个特定的键，如果我们愿意，我们可以使用它来定义要对哪一列进行排序。

sort() 方法就地修改列表，sorted() 内置函数从可迭代对象构建一个新的排序列表。我们还将研究为 sort() 和 sorted() 函数定义可迭代的不同方法。

使用 Python 中的 sort() 函数按列号对二维数组进行排序

为了按列号对数组进行排序，我们必须在函数 sort() 中定义 key，例如，

lst = [["John", 5], ["Jim", 9], ["Jason", 0]]

lst.sort(key=lambda x: x[1])
print(lst)

输出：

[['Jason', 0], ['John', 5], ['Jim', 9]]

出于排序原因，应将 key 参数设置为接受单个参数并返回可在排序过程中使用的 key 的函数的值。可以快速执行此策略，因为每个输入记录只调用一次键函数。

一种常用的模式是使用一个或多个对象的索引作为键对复杂对象进行排序。

lst = [
    ("john", "C", 15),
    ("jane", "A", 12),
    ("dave", "D", 10),
]

lst.sort(key=lambda lst: lst[2])
print(lst)

输出：

[('dave', 'D', 10), ('jane', 'A', 12), ('john', 'C', 15)]

在上述 key=lambda lst:lst[2] 的代码中，lst[2] 定义了应该使用哪一列作为排序依据。在我们的例子中，lst 按第三列排序。

使用 Python 中的 sorted() 函数按列号对二维数组进行排序

为了按列号对数组进行排序，我们必须在函数 sorted() 中定义 key，例如，

li = [["John", 5], ["Jim", 9], ["Jason", 0]]

sorted_li = sorted(li, key=lambda x: x[1])
print(sorted_li)

输出：

[['Jason', 0], ['John', 5], ['Jim', 9]]

请注意，sorted() 函数在前面的代码中返回一个新列表，而 sort() 函数替换了原始列表。

key 也可以使用库 operator 中的 itemgetter 来定义。

from operator import itemgetter

lst = [
    ("john", "C", 15),
    ("jane", "A", 12),
    ("dave", "D", 10),
]

sorted_lst = sorted(lst, key=itemgetter(1))
print(sorted_lst)

输出：

[('jane', 'A', 12), ('john', 'C', 15), ('dave', 'D', 10)]