Pandas:生成一个时间序列,其中填充了一年中的最后一天

Pandas: Generate a timeseries filled with the last day of the year

假设我有一个 Pandas 不规则间隔的时间序列。

2010-01-04   88.82 
2010-11-29   90.70 
2010-12-01   90.09  
2011-02-26   90.10 
2011-08-01   90.55  
2011-09-21   89.50  
2012-04-01   89.06 
2012-04-30   90.22  
2012-05-03   90.21

我想从索引创建另一个时间序列,其中列由一年的最后一个日期填充。因此,对于 2010 年的日期,它将显示 2010-12-01,对于 2011 年的日期,它将显示 2011-09-21,等等。所需的输出是

2010-01-04   2010-12-01
2010-11-29   2010-12-01
2010-12-01   2010-12-01 
2011-02-26   2011-09-21
2011-08-01   2011-09-21 
2011-09-21   2011-09-21  
2012-04-01   2012-05-03
2012-04-30   2012-05-03 
2012-05-03   2012-05-03

我可以提取索引并按年份对它们进行分组。

end_dates=[]
df_idx = df.index
year_df = df_idx.groupby(df_idx.year)
for yr in year_df.keys():
    end_dates.append(max(year_df[yr]))

这给了我一个年度结束日期列表。但是我如何将这些结束日期关联回原始索引以达到所需的输出?

确保您的索引是 datetimeindex 对象。

如果你有 pandas 个系列,你可以使用这个:

s.to_frame().assign(end_dates=s.groupby(s.index.year).transform(lambda x: x.index.max()))

或者如果您已经有数据框:

df.assign(end_dates=df.groupby(df.index.year)['A'].transform(lambda x: x.index.max()))

输出:

                1  end_dates
0                           
2010-01-04  88.82 2010-12-01
2010-11-29  90.70 2010-12-01
2010-12-01  90.09 2010-12-01
2011-02-26  90.10 2011-09-21
2011-08-01  90.55 2011-09-21
2011-09-21  89.50 2011-09-21
2012-04-01  89.06 2012-05-03
2012-04-30  90.22 2012-05-03
2012-05-03  90.21 2012-05-03