python - 数据透视表总和年份

Question

我创建了

cvv=dd.pivot_table('quantity',index='date',aggfunc=sum)

date   quantity
2012  555
2013  397 
2014  876 
2015  129 
2016  984 
2017    5

我只想对数据框中最近 2 年的数据进行求和(数量)。

这样写容易吗？

Answer 1

我们可以首先执行过滤，然后选择该列，并对其调用 sum，例如:

cvv[cvv['date'] >= 2015]['quantity'].sum()
#   \_________ ________/\_____ ____/ \_ _/
#             v               v        v
#           filter     select column   aggregate

由于date列是索引，我们还可以使用.loc[2015:2017]来获取列表行，然后对其进行聚合:

cvv.loc[2015:2017]['quantity'].sum()
#   \____ _______/\_____ ____/ \_ _/
#        v              v        v
#      filter   select column   aggregate