【Python】遍歷文件夾中的文件名，將特定后綴的文件名寫到Excel表中

代碼實現功能：將指定文件夾中具有特定后綴的文件名寫到Excel表中。

#!/usr/bin/python
import os,sys
import xlwt

path = "/xxx/yyy/zzz/"
dirs = os.listdir(path)

write =xlwt.Workbook()
sheet = write.add_sheet('sheet_name')
i = 0

for file in dirs:
    if os.path.splitext(file)[1]=='.目標后綴名':
        sheet.write(i,0,file)
        i+=1

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