如何基于另外两个保存日期的列创建pandas DataFrame列？

import pandas as pd

# 创建包含日期的DataFrame
data = {'date1': ['2022-01-01', '2022-01-02', '2022-01-03'],
        'date2': ['2022-01-04', '2022-01-05', '2022-01-06']}
df = pd.DataFrame(data)

# 将日期列转换为pandas的日期时间格式
df['date1'] = pd.to_datetime(df['date1'])
df['date2'] = pd.to_datetime(df['date2'])

# 计算两个日期的差值，并创建新的列
df['date_diff'] = df['date2'] - df['date1']

# 打印DataFrame
print(df)

       date1      date2 date_diff
0 2022-01-01 2022-01-04    3 days
1 2022-01-02 2022-01-05    3 days
2 2022-01-03 2022-01-06    3 days