【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行：調用os.path.splitext(file)函數判斷文件的後綴名，該函數將參數拆分爲文件名+擴展名，[0]爲文件名，[1]爲擴展名。

 

# 寫Excel文件流程

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')將以上內容進行保存，務必要有這一行。