Python實現自動整理表格的範例程式碼

2023-03-02 18:00:54

前言

今天，在工作的時候，我的美女同事問我有沒有辦法自動生成一個這樣的表格：

第一列是院校+科目，第二列是年份，第三列是數量。

這張表格是基於這一資料夾填充的，之前要一個資料夾一個資料夾開啟然後手動填寫年份和數量

手動整理需要耗費較長時間，於是我便開發了一個 Python 程式用來自動生成歸納表格

利用正規表示式+OS庫+openpyxl生成真題年份歸納表格

原理

第一步，遍歷資料夾下的所有檔案和子資料夾的名稱，並獲取子資料夾下的檔案的年份資訊和數量資訊

第二步，將年份資訊進行格式化，連續的年份取最小值和最大值，並用“-”連線，單獨的年份單獨提取出，並用頓號連線

第三步，寫入資料到Excel中

目標實現

遍歷檔案，新建資料存放的List

path=os.getcwd()
file_list=list(os.walk(path))
infomation=[]
yearList=[]

獲取資訊

 if '/' in path:
  infomation.append(file_list[i][0].replace(path+'/',''))
 elif '\' in path:
  infomation.append(file_list[i][0].replace(path+'\',''))
 totalNum=len(file_list[i][2])
 for j in range (0,len(file_list[i][2])):
  year=re.findall(r'd{4}',file_list[i][2][j])
  yearList.append(int(year[0]))
 yearList.sort()

年份資訊格式化

for i in range(len(yearList)):
  if not res:
   res.append([yearList[i]])
  elif yearList[i-1]+1==yearList[i]:
   res[-1].append(yearList[i])
  else:
   res.append([yearList[i]])
 y=[]
 for m in range (0,len(res)):
  if(max(res[m])==min(res[m])):
   y.append(str(max(res[m])))
  else:
   y.append(str(min(res[m]))+'-'+str(max(res[m])))
 yearInfo="、".join(y)

儲存資料並輸出到Excel中

infomation.append(yearInfo)
 infomation.append(totalNum)
 print(infomation)
 ws.append(infomation)
 wb.save('表格.xlsx')
 infomation=[]
 yearList=[]

最終的完整程式碼如下

import os
import re
from openpyxl import load_workbook
wb=load_workbook('表格.xlsx')
ws=wb.active
path=os.getcwd()
file_list=list(os.walk(path))
infomation=[]
yearList=[]
for i in range (1,len(file_list)):
 if '/' in path:
  infomation.append(file_list[i][0].replace(path+'/',''))
 elif '\' in path:
  infomation.append(file_list[i][0].replace(path+'\',''))
 totalNum=len(file_list[i][2])
 for j in range (0,len(file_list[i][2])):
  year=re.findall(r'd{4}',file_list[i][2][j])
  yearList.append(int(year[0]))
 yearList.sort()
 res=[]
 for i in range(len(yearList)):
  if not res:
   res.append([yearList[i]])
  elif yearList[i-1]+1==yearList[i]:
   res[-1].append(yearList[i])
  else:
   res.append([yearList[i]])
 y=[]
 for m in range (0,len(res)):
  if(max(res[m])==min(res[m])):
   y.append(str(max(res[m])))
  else:
   y.append(str(min(res[m]))+'-'+str(max(res[m])))
 yearInfo="、".join(y)
 infomation.append(yearInfo)
 infomation.append(totalNum)
 print(infomation)
 ws.append(infomation)
 wb.save('表格.xlsx')
 infomation=[]
 yearList=[]

執行效果

好啦，程式不復雜，不過卻大大提高了工作效率，不得不說，Python真棒！

到此這篇關於Python實現自動整理表格的範例程式碼的文章就介紹到這了,更多相關Python自動整理表格內容請搜尋it145.com以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援it145.com！

Python實現自動整理表格的範例程式碼

目錄

前言

原理

目標實現

執行效果

熱門文章