Python元組拆包撿到8倍鏡快準狠

dongfanger發表於2021-02-19

元組拆包

元組是不可變列表,列表是通過索引取值的,元組也是:

tuple_test = (1, 2, 3)
a = tuple_test[0]
b = tuple_test[1]
c = tuple_test[2]

但Python是出了名的一行程式碼解決問題,元組拆包就是精髓技術之一:

a, b, c = tuple_test
print("%s %s %s" % tuple_test)

把元組一一對應拆出來,就叫做元組拆包。拆包有個要求,元組中的元素數量必須跟接受這些元素的空擋數一致,否則會報錯:

tuple_test = (1, 2, 3)
a, b = tuple_test  # ValueError: too many values to unpack (expected 2)

_佔位符

使用_佔位符可以解決這個問題:

tuple_test = (1, 2, 3)
a, b, _ = tuple_test

這樣就只獲取到部分資料了,這在取函式返回值時特別有用,比如:

import os

_, filename = os.path.split("/home/dongfanger/.ssh/idrsa.pub")
print(filename)  # "idrsa.pub"

*字首

當返回值特別多時,_佔位符寫起來麻煩,可以用*來處理剩下的元素:

>>> a, b, *rest = range(5)
>>> a, b, *rest
(0, 1, [2, 3, 4])

注意rest是個列表,如果沒有足夠元素,會返回空列表:

>>> a, b, *rest = range(2)
>>> a, b, *rest
(0, 1, [])

*字首變數能放在任意位置,比如,放在中間:

>>> a, *body, c, d = range(5)
>>> a, body, c, d
(0, [1, 2], 3, 4)

放在前面:

>>> *head, b, c, d = range(5)
>>> head, b, c, d
([0, 1], 2, 3, 4)

實在是妙啊。

*還有一個作用,把元組拆開作為函式引數:

>>> divmod(20, 8)
(2, 4)
>>> t = (20, 8)
>>> divmod(*t)
(2, 4)

經典寫法*args就是這個道理。

巢狀元組拆包

巢狀元組是指元組中有元組,比如(1, 2, 3, (4, 5)),對於巢狀元組,你可能會想要拆兩遍:

tuple_nest_test = (1, 2, 3, (4, 5))
a, b, c, d = tuple_nest_test
x, y = d
print(a, b, c, x, y)

實際上能一步到位:

tuple_nest_test = (1, 2, 3, (4, 5))
a, b, c, (x, y) = tuple_nest_test
print(a, b, c, x, y)

交換兩個變數的值

元組拆包提供了語法糖,對於交換兩個變數的值的常規寫法:

temp = a
a = b
b = temp

可以切換為優雅寫法:

b, a = a, b

具名元組

元組很像資料庫表記錄,除了沒有表名和欄位名,collections.namedtuple具名元組補償了這個缺憾,它是一個工廠函式,可以用來構建一個帶欄位名的元組和一個有名字的類,比如:

import collections

# 定義
Card = collections.namedtuple("Card", ["rank", "suit"])
# 初始化
card_test = Card("J", "hearts")
# 使用
print(card_test.rank)  # J
print(card_test[1])  # hearts

Card是表名,有兩個表欄位rank和suit。

定義具名元組需要2個引數,第1個引數是類名,第2個引數是欄位名,既可以是可迭代物件(如列表和元組),也可以是空格間隔的字串:

Card = collections.namedtuple("Card", ("rank", "suit"))
Card = collections.namedtuple("Card", "rank suit")

初始化時以一串引數形式傳入建構函式:

card_test = Card("J", "hearts")

既可以通過.運算子,也可以用索引來取值:

print(card_test.rank)
print(card_test[1])

這個帶名字的元組,對除錯程式有很大幫助。

列表與元組

元組是不可變列表,它們就像雙胞胎,長相類似,內在性格卻有不同:

黃色列表獨有,紅色元組特有,元組竟然還多了個s.__getnewargs__()方法!從表中可以清楚地看到,除了跟增減元素相關的方法之外,元組支援列表的其他所有方法。

列表也能拆

既然列表和元組是孿生兄弟,那必然也有共同技能:

list_test = [1, 2, 3]
a, b, c = list_test
>>> divmod(20, 8)
(2, 4)
>>> t = [20, 8]  # 換成列表
>>> divmod(*t)
(2, 4)

列表拆包,也是ok的。

小結

本文介紹了Python神奇操作元組拆包,藉助_佔位符和*字首可以進行更加靈活的取值,具名元組實際用的還比較少,不過看一些原始碼是有的。文章最後比較了列表和元組的差異,列表也能拆包。列表(list)、元組(tuple),以及字串(str),都有一個共同操作:切片。

參考資料:

《流暢的Python》

相關文章