Pandas: суммировать строки DataFrame для данных столбцов
У меня есть следующий DataFrame:
import pandas as pd
df = pd.DataFrame({'a': [1,2,3], 'b': [2,3,4], 'c':['dd','ee','ff'], 'd':[5,9,1]})
Я хотел бы добавить столбец "e", который является суммой столбцов "a", "b" и "d".
Переходя через форумы, я думал, что что-то вроде этого будет работать:
df['e'] = df[['a','b','d']].map(sum)
Но нет!
Я хотел бы реализовать операцию, имеющую список столбцов ['a','b','d'] и df в качестве входов.
Ответы
Ответ 1
Вы можете просто sum и установить param axis=1 для суммирования строк, это не будет игнорировать числовые столбцы:
In [91]:
df = pd.DataFrame({'a': [1,2,3], 'b': [2,3,4], 'c':['dd','ee','ff'], 'd':[5,9,1]})
df['e'] = df.sum(axis=1)
df
Out[91]:
a b c d e
0 1 2 dd 5 8
1 2 3 ee 9 14
2 3 4 ff 1 8
Если вы хотите просто суммировать определенные столбцы, вы можете создать список столбцов и удалить те, которые вас не интересуют:
In [98]:
col_list= list(df)
col_list.remove('d')
col_list
Out[98]:
['a', 'b', 'c']
In [99]:
df['e'] = df[col_list].sum(axis=1)
df
Out[99]:
a b c d e
0 1 2 dd 5 3
1 2 3 ee 9 5
2 3 4 ff 1 7
Ответ 2
Если у вас есть всего несколько столбцов, вы можете написать:
df['e'] = df['a'] + df['b'] + df['d']
Это создает новый столбец e со значениями:
a b c d e
0 1 2 dd 5 8
1 2 3 ee 9 14
2 3 4 ff 1 8
Для более длинных списков столбцов предпочтительным является ответ EdChum.
