python - 将两个 csv 文件的值和结果相加到一个新文件中

Question

我需要一些帮助，但不知道如何开始。

基本上，我有两个 csv 输入文件(来自 2 个不同的节点)，并且希望将这些值相加，并将结果发送到 elasticsearch..

非常感谢任何帮助...

CSV1

node    link    rate-in rate-out
node1   link1   10  20
node1   link2   30  50
node1   link3   40  60

CSV2

node    link    rate-in rate-out
node2   link1   20  10
node2   link2   50  70
node2   link3   80  40

结果-->创建elasticsearch索引

node    link    rate-in rate-out
allnode link1   30  30
allnode link2   80  120
allnode link3   120 100

谢谢!

Answer 1

将 CSV 作为数据帧导入 python 后

import pandas
df1 = pandas.DataFrame({'node':['node1', 'node1', 'node1'], 'link': ['link1', 'link2', 'link3'], 'rate-in': [10, 30, 40], 'rate-out': [20, 50, 60]})
df2 = pandas.DataFrame({'node':['node2', 'node2', 'node2'], 'link': ['link1', 'link2', 'link3'], 'rate-in': [20, 50, 80], 'rate-out': [10, 70, 40]})

result = pandas.concat([df1, df2], axis=0).groupby('link').agg({'node': lambda x : 'allNodes', 'rate-in': 'sum', 'rate-out': 'sum'}).reset_index(drop=False)
result

输出:

link    node    rate-in rate-out
0   link1   allNodes    30  30
1   link2   allNodes    80  120
2   link3   allNodes    120 100