Сохранить результат умножения на существующий массив

Question

Рассмотрим следующий код:

a = numpy.array([1,2,3,4]) 
b = numpy.array([5,6,7,8]) 
# here a new array (b*2) will be created and name 'a' will be assigned to it 
a = b * 2 

Таким образом, можно NumPy написать результат b*2 непосредственно в памяти уже выделено для a, без выделения нового массива?

Answer 1

Да, это возможно, - вам нужно использовать np.multiply с out параметром:

np.multiply(b, 2, out=a) 

Массив a теперь заполнен с результатом b * 2 (и никакой новая память не была выделена для удержания выхода из функция).

Все функции ufuncs NumPy имеют параметр out, что особенно удобно при работе с большими массивами; это помогает свести потребление памяти к минимуму, позволяя повторно использовать массивы. Единственное предостережение в том, что массив должен иметь правильный размер/форму, чтобы удерживать вывод функции.