将数据帧与数组合并?

时间:2015-02-09 20:06:01

标签: python arrays pandas

我希望你能提供一些指导 - 我在Python v2.7中使用Pandas库编写脚本。

脚本的一部分合并了两个数据框 - 一个用于收入,另一个用于性能数据。这些DF都有每日条目,并通过ID列链接。

Performance Dataframe:

     RevID         Date       PartnerName        Performance        Revenue
     1,2           1/2/2015   Johndoe            0.02               0.00
     1             2/2/2015   Johndoe            0.12               0.00
     4             3/2/2015   Johndoe            0.10               0.00

请注意' 1,2'上面一行中指的是需要加在一起的两个ID

收入数据框:

     RevID     Date      Revenue
     1         1/2/2015  24000.00
     2         1/2/2015  25000.00
     1         2/2/2015  10000.00
     4         3/2/2015  94000.00

我的问题是,如何在这两行中执行合并,同时考虑到Performance DF中有时会有一个逗号分隔值(如数组),需要从中查找两个相应的收入行收入DF在一起 - 和日期。

例如,我将如何处理此问题,以便最终表格显示为:

     RevID         Date       PartnerName        Performance        Revenue
     1,2           1/2/2015   Johndoe            0.02               49000.00
     1             2/2/2015   Johndoe            0.12               10000.00
     4             3/2/2015   Johndoe            0.10               94000.00

请注意,第一行中的收入已与RevID 1和2的值一起添加。 在这一点上,任何帮助都会很棒!

1 个答案:

答案 0 :(得分:1)

我只是欺骗这些数据,然后逗号的问题就消失了:

In [11]: res = pd.concat([df.iloc[i] for val, i in g.groups.items() for v in val.split(',')], ignore_index=True)

In [12]: res['RevID'] = sum([val.split(',') for val in g.groups], [])

并确保RevID是数字而不是字符串:

In [13]: res['RevID'] = res['RevID'].convert_objects(convert_numeric=True)

In [14]: res
Out[14]:
  RevID      Date PartnerName  Performance  Revenue
0     1  2/2/2015     Johndoe         0.12        0
1     1  1/2/2015     Johndoe         0.02        0
2     2  1/2/2015     Johndoe         0.02        0
3     4  3/2/2015     Johndoe         0.10        0

这样你可以合并,你基本上就在那里:

In [21]: res.merge(df2, on=['RevID', 'Date'])
Out[21]:
   RevID      Date PartnerName  Performance  Revenue_x  Revenue_y
0      1  2/2/2015     Johndoe         0.12          0      10000
1      1  1/2/2015     Johndoe         0.02          0      24000
2      2  1/2/2015     Johndoe         0.02          0      25000
3      4  3/2/2015     Johndoe         0.10          0      94000

注意:您可能希望在合并之前删除0 Revenue列(然后您不需要指定on)。

如果你想引用一个orignal ID(一些独特的东西),那么你可以将其组合并将收入相加,得到你想要的框架......