как сложить столбцы в pandas
Ответы
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