帶有假日日曆的熊貓日期補償

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