如何使用 Pandas dataframe 从 4 月开始计算财政年度的年初至今（YTD）？

Question

我想使用 Pandas dataframe 计算一个财政年度的 YTD。 我使用下面的代码来找到它。但是我从一月份开始得到 YTD。

report_table['ytd_sales'] = report_table.groupby(['year','month', 'planttype', 'market', 'product'])['sales'].cumsum()

谁能帮我计算从四月到三月（财政年度）的年初至今。

Answer 1

通过numpy.where创建financial_year年度列：

report_table['financial_year'] = np.where(report_table['month'] > 3,  
                                          report_table['year']+1, 
                                          report_table['year'])

或month和year列创建日期时间，然后转换为financial_year ：

report_table['financial_year']=(pd.to_datetime(report_table[['month','year']].assign(day=1))
                                  .dt.to_period('Q-MAR')
                                  .dt.qyear)

---

report_table['ytd_sales'] = report_table.groupby(['financial_year', 'planttype', 'market', 'product'])['sales'].cumsum()

Answer 2

利用 -

report_table['Fiscal Year'] = report_table['Date'].dt.to_period('Q-MAR').dt.qyear
report_table['ytd_sales'] = report_table.groupby(['Fiscal Year',month, 'planttype', 'market', 'product'])['sales'].cumsum()

例子-

import pandas as pd
dates = {'Date': ['1/1/2021', '1/31/2021', '2/1/2021', '2/28/2021', '3/1/2021', '3/31/2021', '4/1/2021', '4/30/2021', '5/1/2021', '5/31/2021', '6/1/2021']}
df = pd.DataFrame(dates)
df['Date'] = pd.to_datetime(df['Date'])
print(df)

df['Fiscal Year'] = df['Date'].dt.to_period('Q-MAR').dt.qyear

参考链接- https://datagy.io/pandas-fiscal-year/