NGSIM中同一车辆的数据集提取——pandas

直接上代码：

降低了时间复杂度O(N)，提取速度大大加快

# -*- coding: utf-8 -*-
"""
用于处理将NGSIM数据集中的车辆数据
将不同车辆的信息存储到独立的sheet中
"""

import pandas as pd

filename = 'E:/研究文件及讨论汇报/博弈论决策小组讨论资料/human_like论文数据集/trajectories-0750am-0805am.csv'   # 改为处理文件的位置
data = pd.read_csv(filename)

rows = data.shape[0]  # 获取行数 shape[1]获取列数
vehicles = list(set(data["Vehicle_ID"]))
print(vehicles)

writer = pd.ExcelWriter(r'trajectories-0750am-0805am_车辆信息1020.xlsx')  # 改为输出表格的信心
row = 0
for vehicle in vehicles:
    vehicle_excel = pd.DataFrame()
    while row <= rows :
        if data["Vehicle_ID"][row] > vehicle :
            break
        if data["Vehicle_ID"][row] == vehicle:
            vehicle_excel = pd.concat([vehicle_excel, data.iloc[[row], :]], axis=0, ignore_index = True)
            row += 1

    vehicle_excel.to_excel(writer, sheet_name = str(vehicle), index=False)  # 将每个车存成一个新sheet
    print(str(vehicle)+'号车提取完成')

writer.save()