Python:通过另一列中的每个唯一行条目对数据框中的多个列求和

时间:2017-09-15 11:04:26

标签: python dataframe sum

如何通过使用另一列中的每个唯一条目(例如,通过列C2)来对多个列(例如,列C4,C5和C6)求和。

例如,我想创建一个新的数据框,它将通过导入和导出折叠列C2,并显示C4,C5和C6的总和(最好是删除其他列C1和C3)。

Sample Table

1 个答案:

答案 0 :(得分:0)

您可以使用pandas groupby函数

执行此操作
df = pd.DataFrame([['A', 'Import', 'Argentina', 1, 1, 1],
                   ['B', 'Import', 'Brazil', 2, 2, 2],
                   ['C', 'Export', 'UJ', 3, 3, 3],
                   ['D', 'Export', 'US', 4, 4, 4],
                   ['A', 'Export', 'Canada', 5, 5, 5],
                   ['B', 'Export', 'Russia', 6, 6, 6],
                   ['C', 'Import', 'China', 7, 7, 7],
                   ['D', 'Import', 'India', 8, 8, 8]], 
                  columns=['C1', 'C2', 'C3', 'C4', 'C5', 'C6'])

results = df.groupby("C2").sum()

print(results)

哪个会给你

        C4  C5  C6
C2                
Export  18  18  18
Import  18  18  18