哈嘍大家好，我是咸魚。

我們在使用  sorted() 或 map() 函數(shù)的時候，都會看到里面有一個 key 參數(shù)，其實這個 key 參數(shù)也存在于其他內(nèi)置函數(shù)中（例如 min()、max() 等），那么我們今天就來了解一下 key 參數(shù)的含義以及用途吧！

sorted() 中的 key

我們來看下面這段代碼：

some_numbers = [
    3.14159,
    2.71828,
    6.022e23,
    6.626e-34,
    299_792_458,
    6.674e-11,
    1.61803,
]

reordered_numbers = sorted(some_numbers)

print(*reordered_numbers, sep="\n")

reordered_numbers = sorted(some_numbers) 這一行代碼使用 Python 的內(nèi)置函數(shù) sorted() 對 some_numbers 列表中的數(shù)值進行排序。

*reordered_numbers 將列表中的元素作為參數(shù)傳遞給 print() 函數(shù)，其中解包操作符 * 逐個解包列表中的元素，這相當(dāng)于在括號內(nèi)直接用逗號分隔每個元素。

輸出如下：

6.626e-34
6.674e-11
1.61803
2.71828
3.14159
299792458
6.022e+23

如果我們把要排序的元素由【數(shù)字】改成【名字】，看下會發(fā)生什么：

some_names = [
    "Robert",
    "Ishaan",
    "Max",
    "Trevor",
    "Alexandra",
    "Albert",
    "Christine",
]

reordered_names = sorted(some_names)
print(*reordered_names, sep="\n")

輸出如下：

Albert
Alexandra
Christine
Ishaan
Max
Robert
Trevor

可以看到是按照名字的字母順序進行排序，那如果我們想要根據(jù)名稱的長度來進行排序呢？

也就是說，我想自己定義排序的規(guī)則，這便是 key 參數(shù)的使用場景了。

我們可以把自定義的規(guī)則用函數(shù)的形式表示出來，然后再把函數(shù)名作為  key 參數(shù)的值：

some_names = [
    "Robert",
    "Ishaan",
    "Max",
    "Trevor",
    "Alexandra",
    "Albert",
    "Christine",
]

reordered_names = sorted(some_names, key=len)
print(*reordered_names, sep="\n")

首先列表中 some_names 的每個元素都作為參數(shù)傳遞給函數(shù) len() ，然后 sorted() 使用 len() 返回的值來確定元素的順序。

輸出如下：

Max
Robert
Ishaan
Trevor
Albert
Alexandra
Christine

上面我們說過可以把自定義的規(guī)則用函數(shù)的形式表示出來，然后再把函數(shù)名作為  key 參數(shù)的值。

這里的函數(shù)可以是：

• 內(nèi)置函數(shù)
• 自定義函數(shù)
• 匿名函數(shù)（lambda）

下面是一個自定義函數(shù)的例子：

some_names = [
    "Robert",
    "Ishaan",
    "Max",
    "Trevor",
    "Alexandra",
    "Albert",
    "Christine",
]

def get_number_of_a_s(item):
    return item.lower().count("a")

reordered_names = sorted(some_names, key=get_number_of_a_s)
print(*reordered_names, sep="\n")

函數(shù) get_number_of_a_s() 將輸入字符串轉(zhuǎn)換為小寫，并計算字母 “a” 的出現(xiàn)次數(shù)。該函數(shù)返回此計數(shù)，用于 sorted() 確定新列表中元素的順序。輸出如下：

Robert
Trevor
Christine
Max
Albert
Ishaan
Alexandra

由于 sorted() 通過按升序?qū)?shù)值進行排序來處理數(shù)值，因此沒有“a”的名稱首先出現(xiàn)，因為 .count("a") 這些名稱的返回 0。

"Max" 并且是 "Albert" 下一個，因為它們包含一個出現(xiàn)的 “a”。"Max" 列在最前面，因為它在原始列表中出現(xiàn)之前 "Albert" 。接下來是出現(xiàn)兩次和三次 “a” 的名稱。

下面是一個 lambda 函數(shù)的例子：

some_names = [
    "Robert",
    "Ishaan",
    "Max",
    "Trevor",
    "Alexandra",
    "Albert",
    "Christine",
]

reordered_names = sorted(
    some_names,
    key=lambda item: item.lower().count("a"),
)
print(*reordered_names, sep="\n")

list.sort() 中的 key

sorted()：

• 返回一個新的排序列表
• 不會修改原始列表
• 可以給定一個自定義的比較函數(shù)

list.sort()：

• 對列表進行原地排序
• 會修改原始列表
• 不能給定一個自定義的比較函數(shù)

我們來看看列表的排序函數(shù) list.sort()，需要注意的是：list.sort() 不像內(nèi)置函數(shù) sorted() 那樣返回一個新的排序列表，而是對原有列表進行排序：

some_names = [
    "Robert",
    "Ishaan",
    "Max",
    "Trevor",
    "Alexandra",
    "Albert",
    "Christine",
]

some_names.sort(
    key=lambda item: item.lower().count("a")
)
print(*some_names, sep="\n")

輸出如下：

Robert
Trevor
Christine
Max
Albert
Ishaan
Alexandra

max() 和 min() 中的 key

不單單 sorted() 和 list.sort() 函數(shù)有 key 參數(shù)，max()\min() 里面也有。比如說我想返回一個隨機列表中的最大值，并自定義了比較規(guī)則：

import random

numbers = [random.randint(1, 50) for _ in range(20)]

print(numbers)

# 輸出出列表 numbers 中的最大值
print(
    max(numbers)
)

print(
    max(
        numbers,
        key=lambda x: sum(int(y) for y in str(x)),
    )
)

輸出如下：

[6, 8, 44, 16, 46, 43, 23, 26, 33, 28, 32, 26, 15, 38, 32, 38, 23, 13, 21, 26]
46
38

其中 key 參數(shù)的值為 lambda 函數(shù)，使用 lambda 函數(shù)來自定義規(guī)則，計算列表中每個元素的各位數(shù)之和，并找出其中的最大值：

• str(x) 將整數(shù)轉(zhuǎn)換為字符串，int(y) 將字符串轉(zhuǎn)換為整數(shù)，然后 sum() 函數(shù)計算了該數(shù)字的各位數(shù)之和。
• max() 函數(shù)根據(jù)這個規(guī)則找到了列表中數(shù)字各位數(shù)之和最大的那個數(shù)字

其他

除此之外，像 heapq 模塊中的nlargest() 和  nsmallest() 函數(shù)、itertools 模塊中的groupby() 函數(shù)也有 key 參數(shù)。

我們來看個例子，使用 itertools.groupby() 函數(shù)，將名字列表 some_names 按照名字長度進行分組，并將分組結(jié)果打印輸出：

import itertools

some_names = [
    "Robert",
    "Ishaan",
    "Max",
    "Trevor",
    "Alexandra",
    "Albert",
    "Christine",
]

output = itertools.groupby(
    some_names,
    key=len,
)
for item, group in output:
    print(item, list(group))

函數(shù) itertools.groupby() 有兩個參數(shù)（第二個參數(shù)是可選的）。第一個參數(shù)是包含數(shù)據(jù)的可迭代對象，第二個參數(shù)是 key。

key=len 表示按照元素的長度進行分組。groupby() 函數(shù)將根據(jù)指定的 key 返回一個迭代器，該迭代器產(chǎn)生一對元素，第一個元素是分組的鍵（這里是名字的長度），第二個元素是對應(yīng)的分組中的元素。

輸出如下：

6 ['Robert', 'Ishaan']
3 ['Max']
6 ['Trevor']
9 ['Alexandra']
6 ['Albert']
9 ['Christine']