highD数据集轨迹提取代码

可以使用以下代码提取HighD数据集中的轨迹：

import pandas as pd

# 读取数据集
df = pd.read_csv('trajectories-0400-0415.csv')

# 按照VehicleID和FrameID排序
df = df.sort_values(['VehicleID', 'FrameID'])

# 将数据按照VehicleID分组
groups = df.groupby('VehicleID')

# 遍历每个VehicleID的数据
for vehicle_id, group in groups:
    # 将轨迹数据保存到文件中
    filename = f'{vehicle_id}.csv'
    group[['Local_X', 'Local_Y']].to_csv(filename, index=False)

这段代码做了以下几件事情：

1. 使用Pandas库读取HighD数据集中的轨迹数据；
2. 按照VehicleID和FrameID排序，确保轨迹数据是按照时间顺序排列的；
3. 将轨迹数据按照VehicleID分组；
4. 遍历每个VehicleID的数据，将其保存到以VehicleID命名的csv文件中。

需要注意的是，这段代码只提取了轨迹的位置信息（Local_X和Local_Y），如果需要提取其他信息，可以将代码中的`['Local_X', 'Local_Y']`替换为需要提取的列名。此外，还可以根据需要进行数据清洗和预处理。