Запись в готовый массив в NumPy
При обычном сложении NumPy строит новый объект, а можно указать, куда положить ответ: тогда заполнится переданный массив, если его форма подходит под результат операции.
Сложим два ряда цен и запишем сумму в заранее подготовленный буфер:
import numpy as np
prices_a = np.array([10, 20, 30])
prices_b = np.array([1, 2, 3])
dest = np.zeros(3)
np.add(prices_a, prices_b, out=dest)
print(dest) # выведет [11. 22. 33.]
Массив dest изменился на месте:
отдельный новый ряд для ответа
не понадобился.
import numpy as np
dest = np.array([100., 100., 100.])
np.add(np.array([5, 6, 7]), np.array([1, 1, 1]), out=dest)
print(dest) # выведет [ 6. 7. 8.]
Создайте ряды цен [4, 5, 6]
и [1, 1, 1], подготовьте
буфер из нулей той же длины,
запишите в него сумму и выведите
буфер.
Возьмите буфер [9., 9., 9.],
сложите в него два ряда оценок
[1, 2, 3] и [10, 20, 30]
и выведите, чем он стал.