我正在努力编写一个函数来计算两个选定年份(2019、2020)之间总收入的相对差异.
以下是我的数据:
order_id | price | quantity| date|
1 30 2 20-03-2022
2 50 4 21-04-2020
3 45 3 20-03-2020
4 200 4 21-04-2019
5 54 2 04-03-2020
6 200 4 15-04-2019
这就是我所做的,但任务是编写一个能够做到这一点的函数.(例如def dif():"
df['year'] = pd.DatetimeIndex(df['date']).year
df['revenue']=df.price*order.quantity #calculates revenue for each line of df
total_revenue = df.groupby(['year'])['revenue'].sum() #this gives me revenue by each year
rev2019 = total_revenue.loc[total_revenue["year"] = 2019 ]
rev2020 = total_revenue.loc[total_revenue["year"] = 2020 ]
dif=(rev2019/rev2020)-1
我想把所有这些都放在一个功能下.如果我想要将"年份"更改为不同的数字,如果此函数可以重复使用,也会很有帮助.