在Python中,a += b并不总是等价于a = a + b

大家经常在一些博客中看到这样的说法:

1
a += 1

等价于

1
a = a + 1

这种说法实际上并不准确。

我们来看一个例子:

1
2
3
4
5
6
7
8
9
10
>>> a = [1, 2, 3]
>>> a += (4,)
>>> a
[1, 2, 3, 4]

>>> a = [1, 2, 3]
>>> a = a + (4,)
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
TypeError: can only concatenate list (not "tuple") to list

这里报错了,说明a += ba = a + b并不是完全等价的。

实际上,这是由于+=会首先调用左边这个对象的__iadd__方法,如果没有__iadd__方法,就会调用__add__方法。但是如果直接使用+号,就会直接调用__add__方法。而对于字符串、数字、浮点数这种不可变对象,他们没有__iadd__方法,所以对他们来说,a += ba = a + b是等价的。

但是列表是一个可变的容器,它内部是有__iadd__这个方法。对于列表来说,它的__iadd__方法的原型如下:

1
2
3
def __iadd__(self, values):
self.extend(values)
return self

这一段代码你可以在这里看到:https://github.com/python/cpython/blob/1b5f9c9653f348b0aa8b7ca39f8a9361150f7dfc/Lib/_collections_abc.py

所以说,当你使用+=连接列表和元组的时候,本质上是列表使用extend把元组的内容添加进去。这样是不会报错的:

1
2
3
4
>>> a = [1, 2, 3]
>>> a.extend((4,))
>>> a
[1, 2, 3, 4]