拆分任意長可迭代對象后賦值給多個變量

問題

若一個可迭代對象的元素個數超過變量個數時，會拋出一個 ValueError，那如何才能從這個可迭代對象中拆分出N個元素出來？

解法

Python的星號表達式可以用來解決這個問題。

比如，你在學習一門課程，在學期末的時候，你想統計下家庭作業的平均成績，但是排除掉第一個和最后一個分數。如果只有四個分數，你可能就直接去簡單的手動賦值，但如果有24個呢？

此時星號表達式閃亮登場。

def drop_first_last_and_get_avg(grades):first, *middle, last = gradesreturn avg(middle)

---

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

>>> 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變量永遠都是列表類型，不管拆分的電話號碼數量是多少(包括0個)。所以，任何使用到phone_numbers變量的代碼就無需做多余的類型檢查去確認它是否是列表類型了。

---

星號表達式也能用在列表的開始部分。

比如，你有一個公司前8個月銷售數據的序列，但是你想看下最近一個月數據和前面7個月的平均值的對比。你可以這樣做：

*trailing_qtrs, current_qtr = sales_record trailing_avg = sum(trailing_qtrs) / len(trailing_qtrs) return avg_comparison(trailing_avg, current_qtr) >>> *trailing, current = [10, 8, 7, 1, 9, 5, 10, 3] >>> trailing [10, 8, 7, 1, 9, 5, 10] >>> current 3

討論

擴展的迭代拆分語法是專門為解壓不確定個數或任意個數元素的可迭代對象而設計的。

通常，這些可迭代對象的元素結構有確定的規則（比如第 1 個元素后面都是電話號碼），星號表達式讓開發人員可以很容易的利用這些規則來解壓出元素來。

而不是通過一些比較復雜的手段去獲取這些關聯的的元素值。

---

值得注意的是，星號表達式在迭代元素為可變長元組的序列時是很有用的。比如，下面是一個帶有標簽的元組序列：

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)

---

星號解壓語法在字符串操作的時候也會很有用，比如字符串的分割。

>>> 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'

---

有時候，你想解壓一些元素后丟棄它們，你不能簡單就使用 * ，但是你可以使用一
個普通的廢棄名稱，比如_或者abc。

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

---

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

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

---

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

>>> def sum(items): ... head, *tail = items ... return head + sum(tail) if tail else head # (head + sum(tail)) if tail else head 加個括號更易理解 ... >>> sum(items) 36

然后，由于語言層面的限制，遞歸并不是Python擅長的。因此，最后那個遞歸演示僅僅是個好奇的探索罷了，對這個不要太認真了。

總結

以上是生活随笔為你收集整理的《Python Cookbook 3rd》笔记（1.2）：拆分任意长可迭代对象后赋值给多个变量的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。