【Python】遍历文件夹中的文件名,将特定后缀的文件名写到Excel表中

代码实现功能:将指定文件夹中具备特定后缀的文件名写到Excel表中。html

 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模块能够遍历文件夹。python

3行:xlwt模块能够写Excel文件,该模块在Linux系统中比较好安装,Windows下好像会比较麻烦。函数

 

# 遍历文件名spa

6行:使用os.listdir()获取目标文件夹包含的文件和文件夹名字的列表,按照字母顺序排列,返回值dirs。参考资料code

13行:调用os.path.splitext(file)函数判断文件的后缀名,该函数将参数拆分为文件名+扩展名,[0]为文件名,[1]为扩展名。htm

 

# 写Excel文件流程blog

8行:xlwt.Workbook()建立一个工做簿get

9行:write.add_sheet('sheet_name')向工做簿中添加一个工做表it

14行:sheet.write(i,0,file)向工做表中的第i行,第0列,写入file变量中的内容(行和列都从0开始算起)class

17行:write.save('file_name.xls')将以上内容进行保存,务必要有这一行。