【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')将以上内容进行保存，务必要有这一行。