Почему Numpy рассматривает a + = b и a = a + b иначе

Question

Является ли следующее поведение numpy преднамеренным или это ошибка?

from numpy import * 

a = arange(5) 
a = a+2.3 
print 'a = ', a 
# Output: a = 2.3, 3.3, 4.3, 5.3, 6.3 

a = arange(5) 
a += 2.3 
print 'a = ', a 
# Output: a = 2, 3, 4, 5, 6 

версия Python: 2.7.2, Numpy версия: 1.6.1

Answer 1

Это намеренно.

Оператор += сохраняет тип массива. Другими словами, массив целых чисел остается массивом целых чисел.

Это позволяет NumPy выполнять операцию += с использованием существующего хранилища массивов. С другой стороны, a=a+b создает новый массив для суммы и восстанавливает a, чтобы указать на этот новый массив; это увеличивает объем памяти, используемой для операции.

Цитирую documentation:

Предупреждение: На месте операции будет выполнять вычисления с использованием точности решается типом данных двух операндов, но беззвучно обратное приведение результата (при необходимости), так он может поместиться обратно в массив. Поэтому для смешанных прецизионных расчетов A {op}= B может отличаться от A = A {op} B. Например, предположим, что a = ones((3,3)). Затем a += 3j отличается от a = a + 3j: в то время как оба они выполняют одинаковые вычисления, a += 3 отбрасывает результат, чтобы он возвращался обратно в a, тогда как a = a + 3j повторно связывает имя a с результатом.

Наконец, если вам интересно, почему a был целый массив, в первую очередь, необходимо учитывать следующее:

In [3]: np.arange(5).dtype 
Out[3]: dtype('int64') 

In [4]: np.arange(5.0).dtype 
Out[4]: dtype('float64') 

Answer 2

@aix совершенно прав. Я просто хотел указать, что это не уникально для numpy. Например:

>>> a = [] 
>>> b = a 
>>> a += [1] 
>>> print a 
[1] 
>>> print b 
[1] 
>>> a = a + [2] 
>>> print a 
[1, 2] 
>>> print b 
[1] 

Как вы можете видеть += изменяет список и + создает новый список. Это значение также имеет значение numpy. + создает новый массив, поэтому он может быть любым типом данных. += изменяет массив inplace, и это непрактично, и желательно, для numpy для изменения типа данных массива при изменении содержимого массива.