一个计算pandas数据帧中速度的函数

import pandas as pd

# 自定义函数，用于计算速度
def calculate_speed(row):
    distance = row['distance']  # 获取距离
    time = row['time']  # 获取时间
    speed = distance / time  # 计算速度
    return speed

# 创建一个包含距离和时间的数据帧
data = {'distance': [100, 200, 300],
        'time': [10, 20, 30]}
df = pd.DataFrame(data)

# 使用apply函数调用自定义函数计算速度
df['speed'] = df.apply(calculate_speed, axis=1)

# 打印结果
print(df)

   distance  time  speed
0       100    10   10.0
1       200    20   10.0
2       300    30   10.0