在 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)]