import calendar import datetime import pandas as pd import xlwings as xw import pandas as pd import repath= r"C:\Users\Thinkpad\Desktop\2022年水文资料.xlsx" #-------------------获取表头文件名称及年份---------------------------------------------- app = xw.App(visible=True, add_book=False) # visible 隐式打开文件 book = app.books.open(path) # 打开 excel sht = book.sheets[0] # 选择 sheet btwjm=sht.range('a1:a1').value #获取表头文件名称 print(btwjm) #获取表头文件名称 nf=btwjm.split('年')[0][4:] #获取表头中的年份值 print(nf) book.close() # 关闭 app.quit() # 退出 #-------------------获取表头文件名称及年份---------------------------------------------- #-------------------获取该年份12个月每个的天数---------------------------------------------- i=0 myts=['','','','','','','','','','','',''] #定义一个空列表用于存放该年每月的天数 my=['','','','','','','','','','','',''] while i < 12:#j=i+1myts[i]=calendar.monthrange(int(nf),i+1)[1] #获取该年12个月每月的天数my[i]=str(i+1)+'月份'#print(myts[i])#print(my[i])i=i+1 #print(myts) #print(my) #-------------------获取该年份12个月每个的天数---------------------------------------------- #--------------------读取所有的sheet后合并成1张表------------------------------------------- df = pd.read_excel(path, sheet_name=None) #获取excel表格的所有sheet名 sheetmc=list(df) #获取excel表格的所有sheet名存入列表中sheetmc #print(sheetmc) i=0 df = pd.DataFrame() #创建一个空的DataFrame while i < len(sheetmc):yf = int(str(sheetmc[i]).split('月')[0]) # 获取当前sheet的月份if sheetmc[i]==my[yf-1]: #判断excel表中是否包含12个月的数据dq = pd.read_excel(path, sheetmc[i])dq = dq.loc[3:myts[yf-1] + 2] # 因原始表中所有月天数都是31,这里需要根据该月的天数获取数据dq = dq.iloc[:, 1:12] # 获取第一列到13列的数据# 对列名称进行重新命名dq.columns = ["雨量", "8时水位", "发电量", "发电用水", "弃水量", "生态供水", "径流水量", "流量", "温度", "库 容", "单耗"]# --------------------配置选取数据的时间段------------------------------------# sd获取第一个sheet中月份+01为开始日期sd = str(nf) + "/" + str(sheetmc[i]).split('月')[0] + "/" + "01"# ed获取最后一个sheet中月份+最后一个sheet中月份的天数为最后日期ed = str(nf) + "/" + str(sheetmc[i]).split('月')[0] + "/" + str(myts[int(str(sheetmc[i]).split('月')[0]) - 1])# 设置起始时间s_date = datetime.datetime.strptime(sd, '%Y/%m/%d').date()# 设置结束时间e_date = datetime.datetime.strptime(ed, '%Y/%m/%d').date()print(s_date)print(e_date)# freq="D"表示按天,可以按分钟,月,季度,年等date_range = pd.date_range(start=s_date, end=e_date, freq="D")dq.insert(0,'日期',date_range)# 把重新命名的结果重新保存dq["日期"] = dq["日期"].astype("datetime64") # 确保数据格式为日期# 把日期列只保留年月日dq["日期"] = pd.to_datetime(dq["日期"]).dt.date# --------------------设置选取数据的时间段------------------------------------print(dq)df = pd.concat([df, dq])i=i+1df.to_excel('C:\\Users\\Thinkpad\\Desktop\\'+str(nf)+"年水文雨量.xlsx")
下一篇:黑马程序员 Linux 教程