将两列除以两列

Question

我想将两列收入来自 df1 和来自 df2 的收入按 id 和日期分组。

我做了这样的事情，但出现错误。

df.groupby(['id','date']).agg({'ratio': lambda L: df1['revenue'].div(df2['revenue'])})

数据

df1 = pd.DataFrame({'id':['x1', 'x2'],
           'date':['2021-01-02',
           '2021-01-03'],
       'revenue':[50,10]})

df2 = pd.DataFrame({'id':['x2', 'x1', 'x1'],
            'date':['2021-01-03','2021-01-02', '2021-01-01'],
       'revenue':[100,100, 200]})

预计 output

     id     date    ratio

0   x2  2021-01-03      0.5

1   x1  2021-01-02      0.1

2   x1  2021-01-01      0

Answer 1

使用DataFrame.merge按列id和date左连接，然后将缺失值除以0并替换：

s = df2.merge(df1, on=['id','date'], how='left')['revenue_y'].div(df2['revenue'])
df2['ratio']  = s.fillna(0)
print (df2)
   id        date  revenue  ratio
0  x2  2021-01-03      100    0.1
1  x1  2021-01-02      100    0.5
2  x1  2021-01-01      200    0.0