代码实现功能:将指定文件夹中具有特定后缀的文件名写到Excel表中。
1 #!/usr/bin/python
2 importos,sys3 importxlwt4
5 path = "/xxx/yyy/zzz/"
6 dirs =os.listdir(path)7
8 write =xlwt.Workbook()9 sheet = write.add_sheet('sheet_name')10 i =011
12 for file indirs: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')将以上内容进行保存,务必要有这一行。