使用padas dataframe中的for、If语句计算持续时间

import pandas as pd

# 创建一个示例DataFrame
data = {
    'start_time': ['2023-01-01 10:00:00', '2023-01-02 11:00:00', '2023-01-03 12:00:00'],
    'end_time': ['2023-01-01 11:30:00', '2023-01-02 12:45:00', '2023-01-03 14:00:00']
}
df = pd.DataFrame(data)

# 将时间列转换为datetime类型
df['start_time'] = pd.to_datetime(df['start_time'])
df['end_time'] = pd.to_datetime(df['end_time'])

# 创建一个新列来存储持续时间
df['duration'] = None

# 使用for循环和if语句计算持续时间
for index, row in df.iterrows():
    if pd.notnull(row['start_time']) and pd.notnull(row['end_time']):
        df.at[index, 'duration'] = row['end_time'] - row['start_time']

# 使用向量化操作计算持续时间
df['duration'] = df['end_time'] - df['start_time']