Вычисление суммы элементов строки - распространенная операция при работе с данными в Python. Рассмотрим основные методы выполнения этой задачи.
Содержание
1. Суммирование элементов списка
1.1. Использование встроенной функции sum()
| Код | Описание |
| numbers = [1, 2, 3, 4, 5] | Создание списка чисел |
| total = sum(numbers) | Вычисление суммы элементов |
1.2. Сумма с условием
- sum(x for x in numbers if x % 2 == 0)
- Суммирует только четные числа
2. Суммирование строк матрицы
2.1. Для двумерного списка
- matrix = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
- row_sum = [sum(row) for row in matrix]
- Результат: [6, 15, 24]
2.2. С использованием NumPy
| Код | Результат |
| import numpy as np | Импорт библиотеки |
| arr = np.array(matrix) | Создание массива |
| row_sums = arr.sum(axis=1) | Сумма по строкам |
3. Суммирование элементов строки DataFrame
3.1. В библиотеке Pandas
- import pandas as pd
- df = pd.DataFrame(matrix)
- df['row_total'] = df.sum(axis=1)
- Добавляет столбец с суммой строк
4. Специальные случаи
| Тип данных | Решение |
| Строки с числами | sum(map(int, string_numbers.split())) |
| Словари | sum(dictionary.values()) |
Важные замечания
- Функция sum() работает только с числовыми типами
- Для больших массивов данных используйте NumPy или Pandas
- Убедитесь, что в данных нет None или строковых значений
