如何将工作日中的整数列添加/减去日期时间列？

Question

我的数据框如下：

ID	Days of Holiday	First Day of Holiday
A01	3	16/03/2021
B01	10	24/03/2021
C02	3	31/03/2021
D03	2	02/04/2021

我正在想办法创建另一列“假期返回的第一天”。

我尝试使用下面的 iterrow 循环 DF（上面的 DF 是“日历”）：

for i, r in Calendar.iterrows():\
    Calendar["First Day of Return from holiday"] = Calendar["First Day of Holiday"] + pd.tseries.offsets.BDay(n = r["Days of Holiday"])

而且我没有得到正确的输出。

还有什么方法可以推荐给我吗？

基本上，寻找在工作日内将整数列添加/减去同一行的日期时间列的方法。

非常感谢！

Answer 1

你可以使用 lambda 函数：

df['First Day of Return from holiday'] = df.apply(lambda row: row.['First Day of Holiday'] + row.['Days of Holiday'], axis=1)

Answer 2

首先我们将假期日期的第一天转换为datetime 数据类型，将假期数转换为int 数据类型，然后我们以零开始新列。我们可以使用 BDay 函数迭代并将值分配给新列的每一行以获取工作日。然后我们可以将日期转换回我们需要的日期格式。

但是，日期将被转换为标准格式，可以使用.dt.strftime('%d/%m/%Y') 转换回所需的格式

import pandas as pd
from pandas.tseries.offsets import BDay

df['First Day of Holiday'] = pd.to_datetime(df['First Day of Holiday'])
df['Days of Holiday'] = df['Days of Holiday'].astype('int')
df['First Day of Return from holiday'] = [0]*len(df.index)
for i, r in df.iterrows():
    df.loc[i, 'First Day of Return from holiday'] = r['First Day of Holiday'] + BDay(n=r['Days of Holiday'])
df['First Day of Holiday'] = df['First Day of Holiday'].dt.strftime('%d/%m/%Y')
df['First Day of Return from holiday'] = pd.to_datetime(df['First Day of Return from holiday'])
df['First Day of Return from holiday'] = df['First Day of Return from holiday'].dt.strftime('%d/%m/%Y')
df

输出