快捷導(dǎo)航

Python?中?key?參數(shù)的含義及用法小結(jié)

更新時(shí)間：2023年12月26日 08:59:52 作者：咸魚(yú)Linux運(yùn)維

我們?cè)谑褂?sorted()?或?map()?函數(shù)的時(shí)候，都會(huì)看到里面有一個(gè)?key?參數(shù)，其實(shí)這個(gè)?key?參數(shù)也存在于其他內(nèi)置函數(shù)中（例如?min()、max()?等），那么我們今天就來(lái)了解一下?key?參數(shù)的含義以及用途吧,需要的朋友可以參考下

sorted() 中的 key

我們來(lái)看下面這段代碼：

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() 對(duì) some_numbers 列表中的數(shù)值進(jìn)行排序

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

輸出如下：

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

如果我們把要排序的元素由【數(shù)字】改成【名字（字符串）】，看下會(huì)發(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ìn)行排序，那如果我們想要根據(jù)名稱的長(zhǎng)度來(lái)進(jìn)行排序呢？

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

我們可以把自定義的規(guī)則用函數(shù)的形式表示出來(lái)，然后再把函數(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 的每個(gè)元素都作為參數(shù)傳遞給函數(shù) len() ，然后 sorted() 使用 len() 返回的值來(lái)確定元素的順序

輸出如下：

Max
Robert
Ishaan
Trevor
Albert
Alexandra
Christine

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

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

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

下面是一個(gè)自定義函數(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)換為小寫(xiě)，并計(jì)算字母 “a” 的出現(xiàn)次數(shù)。該函數(shù)返回此計(jì)數(shù)，用于 sorted() 確定新列表中元素的順序。輸出如下：

Robert
Trevor
Christine
Max
Albert
Ishaan
Alexandra

由于 sorted() 通過(guò)按升序?qū)?shù)值進(jìn)行排序來(lái)處理數(shù)值，因此沒(méi)有“a”的名稱首先出現(xiàn)，因?yàn)?nbsp;.count("a") 這些名稱的返回 0

"Max" 并且是 "Albert" 下一個(gè)，因?yàn)樗鼈儼粋€(gè)出現(xiàn)的 “a”。 "Max" 列在最前面，因?yàn)樗谠剂斜碇谐霈F(xiàn)之前 "Albert" 。接下來(lái)是出現(xiàn)兩次和三次 “a” 的名稱

下面是一個(gè) 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()

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

list.sort()

對(duì)列表進(jìn)行原地排序
會(huì)修改原始列表
不能給定一個(gè)自定義的比較函數(shù)

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

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() 里面也有

比如說(shuō)我想返回一個(gè)隨機(jī)列表中的最大值，并自定義了比較規(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ù)來(lái)自定義規(guī)則，計(jì)算列表中每個(gè)元素的各位數(shù)之和，并找出其中的最大值：

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

其他

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

我們來(lái)看個(gè)例子，使用 itertools.groupby() 函數(shù)，將名字列表 some_names 按照名字長(zhǎng)度進(jìn)行分組，并將分組結(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() 有兩個(gè)參數(shù)（第二個(gè)參數(shù)是可選的）。第一個(gè)參數(shù)是包含數(shù)據(jù)的可迭代對(duì)象，第二個(gè)參數(shù)是 key

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

輸出如下：