左边是一个车牌列表,每个车牌号对应一个排放阶段,右侧是车牌号对应的数据,现在的问题是:
我需要把右侧的车牌数据文件按照不同的排放阶段分类,如何利用python实现?
已经解决了,代码如下:
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可以做到的。
用到的软件准备一哈
知识点