【Python 秘籍】1.1 解压序列赋值给多个变量

问题

现在有一个包含 N 个元素的元组或者是序列,怎样将它里面的值解压后同时赋值给 N 个变量?

解决方案

任何序列或者可迭代的对象都可以通过一个简单的赋值操作来分解为单独的变量,这个操作被称为序列解包。唯一的要求就是变量的数量必须跟序列元素的数量是一样的。

代码示例:

>>> p = (4, 5)
>>> x, y = p
>>> x
4
>>> y
5

>>> data = ['ACME', 50, 91.1, (2019, 6, 30)]
>>> name, shares, price, date = data
>>> name
'ACME'
>>> date
(2019, 6, 30)

>>> name, shares, price, (year, mon, day) = data
>>> name
'ACME'
>>> year
2019

如果元素的数量不匹配,将得到一个错误提示。

代码示例:

>>> p = (4, 5)
>>> x, y, z = p
Traceback (most recent call last):
  File "<stdin>", line1, in <module>
ValueError: need more than 2 values to unpack

讨论

实际上不仅仅只是元组或列表,只要对象恰好是可迭代的,那么就可以执行解包操作。这包括字符串、文件、迭代器以及生成器。

代码示例:

>>> s = 'Hello'
>>> a, b, c, d, e = s
>>> a
'H'
>>> b
'e'
>>> e
'o'

当做解包操作时,有时候并不知道具体有多少个值需要解包,这时候可以在变量前加星号。例如:

>>> a, b, *c = 'Hello'
>>> c
['l', 'l', 'o']

注意: 前面加了星号的变量最终的值一定是一个列表。

>>> a, b, *c = 'Hel'
>>> c
['l']
>>> a, b, *c = 'He'
>>> c
[]

有时候,你可能只想解压一部分,丢弃其他的值。对于这种情况 Python 并没有提供特殊的语法。
但是你可以使用任意变量名去占位,到时候丢掉这些变量就行了。

代码示例:

>>> data = [ 'ACME', 50, 91.1, (2012, 12, 21) ]
>>> _, shares, price, _ = data
>>> shares
50
>>> price
91.1
>>>

你必须保证你选用的那些占位变量名在其他地方没被使用到。