如何根据第三列的值计算两个时间戳列之间的差值？

import pandas as pd

# 假设有一个包含三列的DataFrame，分别为时间戳列A、时间戳列B和时间单位列C
data = {'A': [1612345678000, 1612345679000, 1612345680000],
        'B': [1612345680000, 1612345681000, 1612345682000],
        'C': ['ms', 'ms', 'ms']}
df = pd.DataFrame(data)

# 解析时间单位列的值
time_unit = df['C'].iloc[0]

# 将时间戳列的值转换为统一的时间单位（毫秒）
df['A'] = pd.to_numeric(df['A'])
df['B'] = pd.to_numeric(df['B'])

if time_unit == 's':
    df['A'] *= 1000
    df['B'] *= 1000
elif time_unit == 'us':
    df['A'] /= 1000
    df['B'] /= 1000

# 计算差值
df['差值'] = df['B'] - df['A']

# 如果需要将差值转换为其他时间单位，可以进行单位转换，例如将毫秒转换为秒
if time_unit == 'ms':
    df['差值'] /= 1000

print(df)