如何正确为周五日期加天？解决任务时长计算偏差与函数报错

问题背景 #

需要计算任务到达时间（DT_APP_ENTERED）与完成的平均时长，但周五到达的任务因周末不执行导致时长计算值偏大。数据示例：

DT_APP_ENTERED  IN_WORK
0     2023-02-17    54.80
1     2023-02-20    10.36
2     2023-02-21    10.68
3     2023-02-26    19.87
4     2023-02-27    11.35
5     2023-02-28    10.68
6     2023-03-01    10.91
7     2023-03-02    14.36
8     2023-03-03    36.46

用户编写的add_days函数执行时触发错误：Truth value of a Series is ambiguous. Use a.empty, a.bool(), a.item(), a.any() or a.all()

def add_days(start_datetime):
    start_datetime = month['DT_APP_ENTERED']
   
    if start_datetime.dt.weekday == 4:
        start_datetime + dt.timedelta(days=2)
    else:
        start_datetime
    return (start_datetime)

错误原因 #

1. 函数内直接将参数替换为整个DataFrame列month['DT_APP_ENTERED']，逻辑混乱；
2. 使用if判断整个Series的布尔结果，pandas无法解析整个布尔序列的真值，因此抛出歧义错误。

正确解决方案 #

1. 确保日期列是datetime类型 #

先将DT_APP_ENTERED转换为datetime格式（如果还不是的话）：

import pandas as pd
import numpy as np
from datetime import timedelta

# 假设你的数据存储在df中
df['DT_APP_ENTERED'] = pd.to_datetime(df['DT_APP_ENTERED'])

2. 矢量化调整日期 #

使用numpy.where或pandas内置的条件方法，对整个列进行批量处理：

方法一：用numpy.where实现条件赋值

# 给周五（weekday=4）的日期加2天，其他日期保持不变
df['ADJUSTED_ENTERED'] = np.where(
    df['DT_APP_ENTERED'].dt.weekday == 4,
    df['DT_APP_ENTERED'] + timedelta(days=2),
    df['DT_APP_ENTERED']
)

方法二：用pandas.Series.mask实现替换

# 仅对周五的日期进行加2天操作
df['ADJUSTED_ENTERED'] = df['DT_APP_ENTERED'].mask(
    df['DT_APP_ENTERED'].dt.weekday == 4,
    df['DT_APP_ENTERED'] + timedelta(days=2)
)

3. 修正时长计算（可选） #

如果IN_WORK是总小时数，直接减去周末的48小时来修正：

df['ADJUSTED_IN_WORK'] = np.where(
    df['DT_APP_ENTERED'].dt.weekday == 4,
    df['IN_WORK'] - 48,
    df['IN_WORK']
)

内容的提问来源于stack exchange，提问作者Maxim Merkushin