首先要对切分后的NGSIM数据集按照路段类型的不同进行分类，数据集中包含了四种不同的路段，”us-101″、”i-80″、”lankershim”和”peachtree”，其中前两个属于高速路段，后两个属于城市路段，大家可以根据自己的研究需要来选择路段进行研究。
1.获取csv文件名
首先，通过查找目标路径下所有csv文件，并将文件名存储在列表中，以便于接下来的使用吗，代码如下所示：
def Csv_List(self,path_exist):
"""
获取目标路径下的csv文件
:param path_exist:
:return:
"""
csv_list = os.listdir(path_exist)
return csv_list
2.创建存储分类后文件的文件夹
在新的路径下创建文件夹，以便于存储分类后的文件，代码如下所示：
def Create_Folder(self,path):
isExists = os.path.exists(path)
try:
if not isExists:
os.makedirs(path)
else:
pass
except FileExistsError:
print("目录已存在")
3.分类筛选
首先，通过循环的方式，访问原存储路径下的csv文件路径，并读取csv文件；其次通过pandas模块中的条件筛选，将不同路段下的数据集筛选出来，并存储到对应的csv文件下，最后将csv文件存储到新建的目标文件夹下即可。代码如下所示：
road_type_df = df[df["Location"] == folder]
road_location_list = list(loc for loc in road_type_df["Location"] )
if road_location_list != []:
road_type_df.to_csv(csv_path_new,index=False)
print(f"保存{folder}_{count}文件成功")
else:
print(f"{csv}文件中没有{folder}路段")
continue
4.按路段类型分类完成后的图示
Original: https://blog.csdn.net/weixin_50632459/article/details/121283831
Author: 小谢@
Title: 专题2：按照路段类型对切分后的数据集进行分类
原创文章受到原创版权保护。转载请注明出处：https://www.johngo689.com/700425/
转载文章受原作者版权保护。转载请注明原作者出处！