1. 程式人生 > 實用技巧 >【Python】遍歷資料夾中的檔名,將特定字尾的檔名寫到Excel表中

【Python】遍歷資料夾中的檔名,將特定字尾的檔名寫到Excel表中

程式碼實現功能:將指定資料夾中具有特定字尾的檔名寫到Excel表中。

 1 #!/usr/bin/python
 2 import os,sys
 3 import xlwt
 4 
 5 path = "/xxx/yyy/zzz/"
 6 dirs = os.listdir(path)
 7 
 8 write =xlwt.Workbook()
 9 sheet = write.add_sheet('sheet_name')
10 i = 0
11 
12 for file in dirs:
13     if os.path.splitext(file)[1]=='.目標字尾名':
14         sheet.write(i,0,file)
15 i+=1 16 17 write.save('file_name.xls')

2行:os模組可以遍歷資料夾。

3行:xlwt模組可以寫Excel檔案,該模組在Linux系統中比較好安裝,Windows下好像會比較麻煩。

6行:使用os.listdir()獲取目標資料夾包含的檔案和資料夾名字的列表,按照字母順序排列,返回值dirs。參考資料

13行:呼叫

8行:xlwt.Workbook()建立一個工作簿

9行:write.add_sheet('sheet_name')向工作簿中新增一個工作表

14行:sheet.write(i,0,file)向工作表中的第i行,第0列,寫入file變數中的內容(行和列都從0開始算起)

17:行:write.save('file_name.xls')將以上內容進行儲存,務必要有這一行。