pandas 混合索引

需求

我有一张表, 列名是最近一个月的日期range, 但是每隔一天就会被新的日期替换. 我需要在某个日期之前的销量权重按0.5, 某个日期之后的销量按0.8, 再求这个月的平均销量. 如何写这个需求?

解决方案

使用混合索引. 我们获取该日期("20210508")列名的下标, 再结合iloc即可

获取下标

df.columns.get_loc("20210508")

获取20210508之后的数据

temp1 = df.iloc[:, df.columns.get_loc("20210508"):]

这样我们就可以将列名和下标进行结合索引了. 是不是很强. 快夸夸博主吧, 博主一直用爱发电.