1.2 解壓可迭代對(duì)象賦值給多個(gè)變量

問題

如果一個(gè)可迭代對(duì)象的元素個(gè)數(shù)超過變量個(gè)數(shù)時(shí)，會(huì)出現(xiàn)”太多解壓值”的異常。那么怎樣才能從這個(gè)可迭代對(duì)象中解壓出N個(gè)元素出來？

解決方案

Python的星號(hào)表達(dá)式可以用來解決這個(gè)問題。比如，你在學(xué)習(xí)一門課程，在學(xué)期末的時(shí)候，你想統(tǒng)計(jì)下家庭作業(yè)的平均成績(jī)，但是排除掉第一個(gè)和最后一個(gè)分?jǐn)?shù)。如果只有四個(gè)分?jǐn)?shù)，你可能就直接去簡(jiǎn)單的手動(dòng)賦值，但如果有24個(gè)呢？這時(shí)候星號(hào)表達(dá)式就派上用場(chǎng)了：

def drop_first_last(grades):
    first, *middle, last = grades
    return avg(middle)

另外一種情況，假設(shè)你現(xiàn)在有一些用戶的記錄列表，每條記錄包含一個(gè)名字、郵件，接著就是不確定數(shù)量的電話號(hào)碼。你可以像下面這樣分解這些記錄：

>>> record = ('Dave', 'dave@example.com', '773-555-1212', '847-555-1212')
>>> name, email, *phone_numbers = record
>>> name
'Dave'
>>> email
'dave@example.com'
>>> phone_numbers
['773-555-1212', '847-555-1212']
>>>

值得注意的是上面解壓出的phone_numbers變量永遠(yuǎn)都是列表類型，不管解壓的電話號(hào)碼數(shù)量是多少(包括0個(gè))。所以，任何使用到phone_numbers變量的代碼就不需要做多余的類型檢查去確認(rèn)它是否是列表類型了。

星號(hào)表達(dá)式也能用在列表的開始部分。比如，你有一個(gè)公司前8個(gè)月銷售數(shù)據(jù)的序列，但是你想看下最近一個(gè)月數(shù)據(jù)和前面7個(gè)月的平均值的對(duì)比。你可以這樣做：

*trailing_qtrs, current_qtr = sales_record
trailing_avg = sum(trailing_qtrs) / len(trailing_qtrs)
return avg_comparison(trailing_avg, current_qtr)

下面是在Python解釋器中執(zhí)行的結(jié)果：

>>> *trailing, current = [10, 8, 7, 1, 9, 5, 10, 3]
>>> trailing
[10, 8, 7, 1, 9, 5, 10]
>>> current
3

討論

擴(kuò)展的迭代解壓語法是專門為解壓不確定個(gè)數(shù)或任意個(gè)數(shù)元素的可迭代對(duì)象而設(shè)計(jì)的。通常，這些可迭代對(duì)象的元素結(jié)構(gòu)有確定的規(guī)則（比如第1個(gè)元素后面都是電話號(hào)碼），星號(hào)表達(dá)式讓開發(fā)人員可以很容易的利用這些規(guī)則來解壓出元素來。而不是通過一些比較復(fù)雜的手段去獲取這些關(guān)聯(lián)的的元素值。

值得注意的是，星號(hào)表達(dá)式在迭代元素為可變長(zhǎng)元組的序列時(shí)是很有用的。比如，下面是一個(gè)帶有標(biāo)簽的元組序列：

records = [
    ('foo', 1, 2),
    ('bar', 'hello'),
    ('foo', 3, 4),
]

def do_foo(x, y):
    print('foo', x, y)

def do_bar(s):
    print('bar', s)

for tag, *args in records:
    if tag == 'foo':
        do_foo(*args)
    elif tag == 'bar':
        do_bar(*args)

星號(hào)解壓語法在字符串操作的時(shí)候也會(huì)很有用，比如字符串的分割。

代碼示例：

>>> line = 'nobody:*:-2:-2:Unprivileged User:/var/empty:/usr/bin/false'
>>> uname, *fields, homedir, sh = line.split(':')
>>> uname
'nobody'
>>> homedir
'/var/empty'
>>> sh
'/usr/bin/false'
>>>

有時(shí)候，你想解壓一些元素后丟棄它們，你不能簡(jiǎn)單就使用*，但是你可以使用一個(gè)普通的廢棄名稱，比如_或者ign。

代碼示例：

>>> record = ('ACME', 50, 123.45, (12, 18, 2012))
>>> name, *_, (*_, year) = record
>>> name
'ACME'
>>> year
2012
>>>

在很多函數(shù)式語言中，星號(hào)解壓語法跟列表處理有許多相似之處。比如，如果你有一個(gè)列表，你可以很容易的將它分割成前后兩部分：

>>> items = [1, 10, 7, 4, 5, 9]
>>> head, *tail = items
>>> head
1
>>> tail
[10, 7, 4, 5, 9]
>>>

如果你夠聰明的話，還能用這種分割語法去巧妙的實(shí)現(xiàn)遞歸算法。比如：

>>> def sum(items):
... head, *tail = items
... return head + sum(tail) if tail else head
...
>>> sum(items)
36
>>>

然后，由于語言層面的限制，遞歸并不是Python擅長(zhǎng)的。因此，最后那個(gè)遞歸演示僅僅是個(gè)好奇的探索罷了，對(duì)這個(gè)不要太認(rèn)真了。