使用Python批量提取指定的站点空气质量数据

import os
import pandas as pd

# 定义相关参数
dataPath = './data' # 数据目录
targets = ['1001A','1002A','1003A','1004A','1005A','1006A','1007A','1008A'] # 目标站点
result = [[] for i in range(len(targets))] # 用于保存结果

# 开始遍历
for filepath in os.listdir(dataPath): # 遍历每个文件夹
    for filename in os.listdir('%s/%s'%(dataPath,filepath)):
        if not filename.endswith('.csv'): # 去重非csv数据文件
            continue
        data = pd.read_csv('%s/%s/%s'%(dataPath,filepath,filename))
        for i in range(0,len(data),15):
            for k in range(len(targets)):
                try:
                    item = {'date':data['date'][i], # 日期
                            'hour':data['hour'][i]} # 小时
                    for j in range(i,i+15):
                        item[data['type'][j]] = data[targets[k]][j]
                    result[k].append(item)
                except:
                    pass
        print('%s处理完毕'%filename)
            
# 保存结果
for i in range(len(targets)):
    pd.DataFrame(result[i]).to_csv('%s.csv'%targets[i],index=False)