python - Pandas 在假期前添加天数

Question

我需要添加一个“next_holiday”列来计算下一个假期的天数。以下是添加列后的预期输出:

SalesDate holiday   next_holiday
2016-12-20  0             5
2016-12-21  0             4
2016-12-22  0             3
2016-12-23  0             2
2016-12-24  0             1
2016-12-25  1             0

Answer 1

这里有两行建议:

temp = df.holiday.iloc[::-1].eq(1).cumsum()   
df['next_holiday'] = temp.groupby(temp).cumcount()