Функция add
Функция add выполняет поэлементное сложение двух массивов
или массива и скалярного значения. Она является частью модуля
numpy и предоставляет удобный способ сложения данных
с возможностью управления параметрами. Первым и вторым аргументами
передаются массивы или скаляры, третьим (необязательным) -
выходной массив, в который будет сохранён результат.
Синтаксис
np.add(x1, x2, out=None, where=True, dtype=None)
Основные параметры:
-
x1,x2- массивы или скалярные значения для сложения; -
out- необязательный массив для сохранения результата; -
where- условие, определяющее, где выполнять операцию; -
dtype- желаемый тип данных результата.
Пример
Выполним сложение двух одномерных массивов:
import numpy as np
arr1 = np.array([1, 2, 3, 4, 5])
arr2 = np.array([10, 20, 30, 40, 50])
res = np.add(arr1, arr2)
print(res)
Результат выполнения кода:
[11 22 33 44 55]
Пример
Сложение массива со скалярным значением:
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
res = np.add(arr, 10)
print(res)
Результат выполнения кода:
[11 12 13 14 15]
Пример
Использование параметра out для сохранения результата в существующий массив:
import numpy as np
arr1 = np.array([1, 2, 3, 4, 5])
arr2 = np.array([10, 20, 30, 40, 50])
res = np.empty(5, dtype=int)
np.add(arr1, arr2, out=res)
print(res)
Результат выполнения кода:
[11 22 33 44 55]
Пример
Сложение двумерных массивов одинаковой формы:
import numpy as np
arr1 = np.array([[1, 2, 3], [4, 5, 6]])
arr2 = np.array([[10, 20, 30], [40, 50, 60]])
res = np.add(arr1, arr2)
print(res)
Результат выполнения кода:
[[11 22 33]
[44 55 66]]