Note: Make sure to remove the duplicate h1 tag from the beginning of the sentence in the actual code.
Библиотека NumPy предоставляет функцию sum для вычисления суммы элементов массива. Функция принимает массив в качестве аргумента и возвращает сумму его элементов.
Вот пример кода, который иллюстрирует использование функции sum:
python
import numpy as np
# Создаем массив
arr = np.array([1, 2, 3, 4, 5])
# Вычисляем сумму элементов массива
sum_of_elements = np.sum(arr)
# Выводим результат
print("Сумма элементов массива:", sum_of_elements)
В данном примере создается одномерный массив arr с элементами [1, 2, 3, 4, 5]. Затем функция sum вызывается с аргументом arr, и результат присваивается переменной sum_of_elements. Наконец, результат выводится на экран.
Вывод программы будет следующим:
Сумма элементов массива: 15
Также, функция sum может использоваться для вычисления суммы элементов по определенной оси (столбцу или строке) в многомерном массиве. Вот пример кода, который показывает это:
python
import numpy as np
# Создаем двумерный массив
arr = np.array([[1, 2, 3], [4, 5, 6]])
# Вычисляем сумму элементов по столбцам
sum_of_columns = np.sum(arr, axis=0)
# Вычисляем сумму элементов по строкам
sum_of_rows = np.sum(arr, axis=1)
# Выводим результаты
print("Сумма элементов по столбцам:", sum_of_columns)
print("Сумма элементов по строкам:", sum_of_rows)
В данном примере создается двумерный массив arr с элементами [[1, 2, 3], [4, 5, 6]]. Затем функция sum вызывается с аргументом arr и указанием оси, по которой требуется вычислить сумму. Результаты присваиваются переменным sum_of_columns и sum_of_rows соответственно. Наконец, результаты выводятся на экран.
Вывод программы будет следующим:
Сумма элементов по столбцам: [5 7 9]
Сумма элементов по строкам: [6 15]