当涉及切片时,Plus equal 给出的答案与分配给显式总和(这是人们期望的答案)不同。是否有一个原因?应该避免加等于吗?
a = np.arange(10)
b = np.arange(10)
a[3:] += a[:-3]
b[3:] = b[3:] + b[:-3]
print a
#[ 0 1 2 3 5 7 9 12 15 18]
print b
#[ 0 1 2 3 5 7 9 11 13 15]
正如 JBernardo 评论+=
的那样,更改数组。
a[3:] += [a:-3]
类似于以下内容:
>>> import numpy as np
>>> a = np.arange(10)
>>>
>>> for i in range(3, 10):
... print('a[{}] ({}) += a[{}] ({})'.format(i, a[i], i-3, a[i-3]))
... a[i] += a[i-3]
... print(' a[{}] -> {}'.format(i, a[i]))
...
a[3] (3) += a[0] (0)
a[3] -> 3
a[4] (4) += a[1] (1)
a[4] -> 5
a[5] (5) += a[2] (2)
a[5] -> 7
a[6] (6) += a[3] (3)
a[6] -> 9
a[7] (7) += a[4] (5) # NOTE: not (4)
a[7] -> 12
a[8] (8) += a[5] (7)
a[8] -> 15
a[9] (9) += a[6] (9)
a[9] -> 18
为避免这种情况,请使用数组的副本:
>>> a = np.arange(10)
>>> a[3:] += np.copy(a[:-3]) # OR np.array(a[:-3])
>>> a
array([ 0, 1, 2, 3, 5, 7, 9, 11, 13, 15])