带有假日日历的熊猫日期补偿

Question

熊猫目前允许您将工作日添加到给定日期datetime.today() + 3*BDay()。我想延长一个营业日的想法，以排除假期和周末的给定日期索引。这可能将一个DateIndex合并到一个偏移量中吗？

Answer 1

目前我认为你需要创建一个自定义的子类。您需要覆盖apply和onOffset方法来考虑您的假期日历。

最终，我们应该在business-X频率中添加一个可选的假期日历参数。我做了一个GitHub的问题，以保持它的轨道：https://github.com/pydata/pandas/issues/2301

Answer 2

的CustomBusinessDay类现在已经合并到大熊猫即将0.12版本，你就可以做一些类似如下：

>>> from pandas.tseries.offsets import CustomBusinessDay 
>>> 
>>> # As an interesting example, let's look at Egypt where 
>>> # a Friday-Saturday weekend is observed. 
>>> weekmask_egypt = 'Sun Mon Tue Wed Thu' 
>>> 
>>> # They also observe International Workers' Day so let's 
>>> # add that as a holiday for a couple of years 
>>> holidays = ['2012-05-01', datetime(2013, 5, 1), np.datetime64('2014-05-01')] 
>>> 
>>> bday_egypt = CustomBusinessDay(holidays=holidays, weekmask=weekmask_egypt) 
>>> dt = datetime(2013, 4, 30) 
>>> print dt + 2 * bday_egypt 
2013-05-05 00:00:00 
>>> 
>>> dts = date_range(dt, periods=5, freq=bday_egypt).to_series() 
>>> print dts 
2013-04-30 2013-04-30 00:00:00 
2013-05-02 2013-05-02 00:00:00 
2013-05-05 2013-05-05 00:00:00 
2013-05-06 2013-05-06 00:00:00 
2013-05-07 2013-05-07 00:00:00 
Freq: C, dtype: datetime64[ns] 
>>> 
>>> print Series(dts.weekday, dts).map(Series('Mon Tue Wed Thu Fri Sat Sun'.split())) 
2013-04-30 Tue 
2013-05-02 Thu 
2013-05-05 Sun 
2013-05-06 Mon 
2013-05-07 Tue 
dtype: object 

HTH