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

Question

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

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

Answer 1

已经解决了，代码如下：

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)