Python:列表也能拆包?

时间:2022-01-10 15:00:34

前几天,微信学习群里有个小伙伴在看书时遇到了这样一个问题,在群里提问,看下图:

Python:列表也能拆包?

这是常用的 matplotlib 库,只是一般我们调用 plot 方法绘图时,不会去关心它的返回值。然而 plt1, = plt.plot(...) 这种写法是什么意思?被赋值的变量和赋值符号之间多了个逗号。

我的确从来没关心过 plot 函数的返回值是什么,所以当我看到这个截图的时候,第一反应是以为返回值有多个,代码只取了第一个,剩余的被忽略了。这种情况一般都是用 _ 做占位符,来忽略不关心的值。

占位符 _ 也可以省略吗?我迅速打开 PyCharm 验证了一把,事实证明,_ 不可以省略,我的想法是错了。

为了查明原因,我去查了 matplotlib 的源码,发现 plot 函数返回的是一个列表,源码如下:

def plot(self, *args, **kwargs):
    scalex = kwargs.pop('scalex', True)
    scaley = kwargs.pop('scaley', True)     if not self._hold:
        self.cla()
    lines = []     kwargs = cbook.normalize_kwargs(kwargs, _alias_map)     for line in self._get_lines(*args, **kwargs):
        self.add_line(line)
        lines.append(line)     self.autoscale_view(scalex=scalex, scaley=scaley)
    return lines

我这才反应过来,这是列表拆包啊。变量后面的逗号,是因为返回的列表中只有一个元素,所以拆包的时候变量后面要加个逗号,标明这是拆包,不是赋值。

代码中,我们用的比较多的是元组拆包,各种书里也多是讲元组拆包,所以忽略了列表也是可以拆包的。看下面的例子:

x, y = [1, 2]
print(x, y)

输出:

1 2

不仅是元组和列表,所有可迭代对象都可以拆包。我们可以自定义一个可迭代对象(只需要实现其协议即 __iter__ 方法即可),来验证一下,代码如下:

class TestClass:
    def __init__(self):
        self._list = range(3)     def __iter__(self):
        for i in self._list:
            yield i x, y, z = TestClass()
print(x, y, z)

输出:

0 1 2