# -*- coding: utf-8 -*-
"""
Created on Sat Oct 13 14:08:19 2018
@author: Administrator
"""
#import re
import os
import pandas as pd
inws=r'G:\QX_data\2002_2017\PRE_CSVaddTtile' #输入路径
outws=r'G:\QX_data\2002_2017\PRE_CSVperSite'#输出路径
site=[51709,51747,51765,51811,51828] #需要用到 四个站点 2002-2017年的数据,每个文件存放四个站点,每年每月每天的数据,需要将个文件中的站点数据合并到一个文件,比如PRE_51828.csv,包含所有这个站点的2002-2017每年每月每天的数据
title=['区站号','纬度','经度','观测场拔海高度','年','月','日','20-8时降水量','8-20时降水量','20-20时累计降水量','20-8时降水量质量控制码','8-20时累计降水量质量控制码','20-20时降水量质量控制码'] #数据表头
#新建存放输出数据的CSV文件
#for s in site:
# K=outws+'\\'+'PRE_'+str(s)+'.csv'
# print(K)
# namelist.append(K)
# sitefile=open(K,'w+')
#sitefile.close()
#print(namelist)
mery=pd.DataFrame() #空DF
files1= os.listdir(inws)#遍历文件夹
final_data=open('G:\\QX_data\\2002_2017\\PRE_CSVperSite\\PRE_51828.csv','w+') #打开存放数据的空文件
for file1 in files1: #文件夹下各文件名
print (file1)
Cname=inws+'\\'+file1
data=pd.read_csv(Cname,encoding='gbk',index_col=1) #读取数据,并将区站号设为索引
for m in site:
for n in data.index:
# print(m,n)
index=0
if str(m)==str(n):
# print (data.ix[n])
if str(n)=='51828': #当索引等于站点
print (data.ix[n])
mery=mery.append(data.ix[n]) #循环添加到一个DF
mery.to_csv(final_data,header=title) #写入文件
final_data.close()