遍历文件夹内文件，并输出指定后缀名的文件

赵乾舟

1. import os
   
2. 文件 = []                     #创建列表存放文件名
   
3. 文件夹 =[]                     #创建列表存放文件夹名
   
4. for i in os.scandir(os.getcwd()):  #遍历列表里的所有文件名
   
5.     if i.is_dir():
   
6.         文件夹.append(i.name)
   
7.     else:
   
8.         文件.append(i.name)
   
9. list = []                       #创建一个空列表，存带关键字的文件名
   
10. for a in 文件:
    
11.     houzhui = a.rfind('.')          #找到后缀名点的位置
    
12.     if (a[houzhui:] == '.xlsx'):        #利用切片拿到后缀名和指定的后缀对比
    
13.         list.append(a)
    
14. 
    
15. print(f'Excle文件一共有{len(list)}个，分别是{list}')

复制代码

赵乾舟

运行结果：

1. Excle文件一共有16个，分别是['jia.xlsx', 'mine.xlsx', 'result.xlsx', 'wenjian.xlsx', '一个测试3.xlsx', '人员爬取.xlsx', '文件名.xlsx', '文件名1.xlsx', '文件名2.xlsx', '测1

复制代码

赵乾舟

第11和12行代码也可以修改为如下代码：

1.     if a.endswith('xlsx'):

复制代码