如何计算每个begin和end的差异- Pandas Python

import pandas as pd

# 创建示例数据
data = {'begin': ['2022-01-01 10:00:00', '2022-01-02 12:00:00', '2022-01-03 15:00:00'],
        'end': ['2022-01-01 12:00:00', '2022-01-02 14:00:00', '2022-01-03 17:00:00']}
df = pd.DataFrame(data)

# 将日期字符串转换为日期时间对象
df['begin'] = pd.to_datetime(df['begin'])
df['end'] = pd.to_datetime(df['end'])

# 计算差异
df['diff'] = df['end'] - df['begin']

# 打印结果
print(df)

                begin                 end            diff
0 2022-01-01 10:00:00 2022-01-01 12:00:00 0 days 02:00:00
1 2022-01-02 12:00:00 2022-01-02 14:00:00 0 days 02:00:00
2 2022-01-03 15:00:00 2022-01-03 17:00:00 0 days 02:00:00