как сложить столбцы в pandas

Аватар пользователя Maksim Litvinov
Maksim Litvinov
06 сентября 2024

Чтобы сложить столбцы в Pandas, можно использовать метод sum() с параметром axis=1. Этот метод вычисляет сумму значений вдоль указанной оси, которая для столбцов является осью 1.

Пример:

import pandas as pd

#создаем DataFrame
data = {'A': [1, 2, 3],
        'B': [4, 5, 6],
        'C': [7, 8, 9]}

df = pd.DataFrame(data)

#сложение столбцов A и B
df['D'] = df[['A', 'B']].sum(axis=1)

print(df)

Результат будет выглядеть следующим образом:

   A  B  C  D
0  1  4  7  5
1  2  5  8  7
2  3  6  9  9
1 0
Познакомьтесь с основами аналитики данных бесплатно