[英]How to set Holidays with Pandas in a timezone-aware way
我有一个按日期时间和每小时频率索引的熊猫数据框(可识别时区):
<class 'pandas.tseries.index.DatetimeIndex'>
[2014-01-01 00:00:00+01:00, ..., 2014-06-30 23:00:00+02:00]
Length: 4933, Freq: None, Timezone: Europe/Rome
我想将我设置的所有假期的工作日设置为6(= Sunday)。 如何以时区感知方式进行操作?
这可能很幼稚,但可以
from pandas.tseries.holiday import Holiday, AbstractHolidayCalendar
from datetime import datetime
class ItalianHolidaysCalendar(AbstractHolidayCalendar):
rules = [Holiday('Pasquetta', month=4, day=21),
Holiday('Liberation day', month=4, day=25),
Holiday('Workers day', month=5, day=1),
Holiday('Republic day', month=6, day=2)
]
cal = ItalianHolidaysCalendar()
holidays_df = pd.DataFrame({})
dd = pd.date_range(datetime(2014,1,1,0), periods=24, freq='H', tz='Europe/Rome')
holidays = cal.holidays(sliceSum.datetime.min().date(), sliceSum.datetime.max().date())
for day in holidays:
dd = dd.append(pd.date_range(day, day + pd.DateOffset(hours=23), freq='H'))
sliceSum['holiday'] = False
sliceSum.loc[sliceSum.index.isin(dd),['holiday']] = True
sliceSum[sliceSum.holiday].head(50)
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.