Python在日期中添加int，只返回工作日日期，忽略周末

Question

我有下面的数据帧

df = pd.DataFrame({'Start_Date': ['2018-08-23', '2018-08-16', '2018-08-24', '2018-08-29'], 'Days': ['3', '1', '3', '0']})

我想创建一个新列"结束日期"。结束日期将为开始日期+天。此外，End_Date应仅为工作日(不包括周末(。

预期退出

Start Date      Days   End_Date
2018-08-23      3      2018-08-28    
2018-08-16      1      2018-08-17
2018-08-24      3      2018-08-29
2018-08-29      0      2018-08-29

我试过使用omz的解决方案，但对我不起作用。我试过用workday函数，但也不起作用。

知道吗？

Answer 1

您可以通过以下方式使用BDay：

from pandas.tseries.offsets import BDay
df['Start_Date'] = pd.to_datetime(df['Start_Date'])
df['End_Date']  = df.apply(lambda x: x['Start_Date'] + BDay(x['Days']), axis=1)

输出：

Days Start_Date   End_Date
0     3 2018-08-23 2018-08-28
1     1 2018-08-16 2018-08-17
2     3 2018-08-24 2018-08-29
3     0 2018-08-29 2018-08-29

Answer 2

您可以使用numpy.busday_offset获取下一个工作日。

In [1]: import numpy as np
In [2]: df['End_Date'] = df.apply(lambda x: np.busday_offset(x[0], x[1]), axis=1)
In [3]: df
Out[3]:
Start_Date Days   End_Date
0  2018-08-23    3 2018-08-28
1  2018-08-16    1 2018-08-17
2  2018-08-24    3 2018-08-29
3  2018-08-29    0 2018-08-29