基于日期范围创建一个新的分类变量列

import pandas as pd

# 读取数据集，假设日期列名为"Date"
data = pd.read_csv("data.csv")

# 将日期列转换为日期格式
data["Date"] = pd.to_datetime(data["Date"])

# 创建新的分类变量列
data["Date Range Category"] = ""

# 定义日期范围
start_date = pd.to_datetime("2022-01-01")
end_date = pd.to_datetime("2022-12-31")

# 遍历数据集
for index, row in data.iterrows():
    date = row["Date"]
    
    # 判断日期是否在范围内
    if start_date <= date <= end_date:
        # 在范围内，赋值分类标签
        data.at[index, "Date Range Category"] = "Category A"
    else:
        # 不在范围内，保持为空值或填充默认值
        data.at[index, "Date Range Category"] = ""

# 输出结果
print(data)