如何在python中创建用于事件研究的虚拟变量

import pandas as pd

# 示例数据集
data = {
    'date': ['2023-01-01', '2023-01-02', '2023-01-03', '2023-01-04'],
    'event': ['none', 'event', 'none', 'none']
}
df = pd.DataFrame(data)

# 定义事件发生的日期
event_date = '2023-01-02'

# 创建虚拟变量
df['event_dummy'] = df['date'].apply(lambda x: 1 if x == event_date else 0)

print(df)

df['date'] = pd.to_datetime(df['date'])

if event_date not in df['date'].values:
    print("指定的事件日期不在数据集中")

event_mapping = {'event1': '2023-01-02', 'event2': '2023-01-03'}
for event_name, event_day in event_mapping.items():
    df[event_name + '_dummy'] = df['date'].apply(lambda x: 1 if x == event_day else 0)