R语言：基于weekday变量条件生成数据框的DateToTake列

嘿，这个日期调整的需求很典型，我给你两个靠谱的实现方式，尤其是考虑到你处理的是大型数据框，我们优先选效率高的方法：

首先第一步，一定要确保你的Date列是datetime类型，不然没法做日期加减操作。如果还没转换的话，先执行这行：

import pandas as pd
df['Date'] = pd.to_datetime(df['Date'])

方法一：用numpy.select（推荐，高效向量化操作） #

对于大型数据框，逐行处理会很慢，所以用numpy.select这种向量化方法是最优解，速度快很多。我们先定义好判断条件和对应的日期调整逻辑：

import numpy as np

# 定义三个判断条件
conditions = [
    # 周一到周五的情况
    df['weekday'].isin(['Monday', 'Tuesday', 'Wednesday', 'Thursday', 'Friday']),
    # 周六的情况
    df['weekday'] == 'Saturday',
    # 周日的情况
    df['weekday'] == 'Sunday'
]

# 对应每个条件的目标日期
choices = [
    df['Date'],
    df['Date'] + pd.Timedelta(days=2),
    df['Date'] + pd.Timedelta(days=1)
]

# 生成最终的DateToTake列
df['DateToTake'] = np.select(conditions, choices, default=df['Date'])

这里加了default参数，以防你的weekday列出现异常值（比如拼写错误），默认返回原日期，避免报错。

方法二：用apply（逻辑直观，适合小数据量） #

如果你的数据量没大到极致，也可以用apply来写，代码逻辑更直白，容易理解：

def calculate_target_date(row):
    if row['weekday'] in ['Monday', 'Tuesday', 'Wednesday', 'Thursday', 'Friday']:
        return row['Date']
    elif row['weekday'] == 'Saturday':
        return row['Date'] + pd.Timedelta(days=2)
    elif row['weekday'] == 'Sunday':
        return row['Date'] + pd.Timedelta(days=1)
    else:
        # 处理异常情况，返回原日期
        return row['Date']

df['DateToTake'] = df.apply(calculate_target_date, axis=1)

额外小技巧 #

如果你的weekday列不是字符串，而是用数字表示的（比如0=周一，4=周五，5=周六，6=周日），或者你不想依赖现有的weekday列，可以直接从Date列提取星期几，这样更准确，还能避免字符串拼写错误：

# 直接从Date列获取星期几数字（0=周一，6=周日）
conditions = [
    df['Date'].dt.weekday <= 4,
    df['Date'].dt.weekday == 5,
    df['Date'].dt.weekday == 6
]

# 后面的choices和之前一样，生成DateToTake列即可
df['DateToTake'] = np.select(conditions, choices, default=df['Date'])

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