python如何实现exel文件分类存储?

左边是一个车牌列表,每个车牌号对应一个排放阶段,右侧是车牌号对应的数据,现在的问题是:

我需要把右侧的车牌数据文件按照不同的排放阶段分类,如何利用python实现?

img

已经解决了,代码如下:


import pandas as pd
import os
# 读取车牌.xlsx文件内容
df = pd.read_excel('./data/车牌.xlsx')
# 创建保存不同排放阶段的文件夹
for stage in df['排放阶段'].unique():
    if not os.path.exists(f'./data/{stage}'):
        os.mkdir(f'./data/{stage}')
# 按照排放阶段分类保存到不同的文件夹中
for file in os.listdir('./data'):
    if file.endswith('.xlsx') and file != '车牌.xlsx':
        filename = os.path.join('./data', file)
        stage = pd.read_excel(filename)['排放阶段'].unique()[0]
        df_tmp = pd.read_excel(filename)
        df_tmp = df_tmp[df_tmp['车牌号'].isin(df[df['排放阶段'] == stage]['车牌号'])]
        df_tmp.to_excel(f'./data/{stage}/{file}', index=False)

最后的分类要显示为什么样的,python可以做到的。