关于零基础30讲用python查找文件夹

dyh55555

我想请教一下如果要查找整个磁盘比如说D盘有什么方法可以做到吗？

1. import os
   
2. 
   
3. def search_file(start_dir, target) :
   
4.     os.chdir(start_dir)
   
5.    
   
6.     for each_file in os.listdir(os.curdir) :
   
7.         ext = os.path.splitext(each_file)[1]
   
8.         if ext in target :
   
9.             vedio_list.append(os.getcwd() + os.sep + each_file + os.linesep) # 使用os.sep是程序更标准
   
10.         if os.path.isdir(each_file) :
    
11.             search_file(each_file, target) # 递归调用
    
12.             os.chdir(os.pardir) # 递归调用后切记返回上一层目录
    
13. 
    
14. start_dir = input('请输入待查找的初始目录：')
    
15. program_dir = os.getcwd()
    
16. 
    
17. target = ['.mp4', '.avi', '.rmvb']
    
18. vedio_list = []
    
19. 
    
20. search_file(start_dir, target)
    
21. 
    
22. f = open(program_dir + os.sep + 'vedioList.txt', 'w')
    
23. f.writelines(vedio_list)
    
24. f.close()
    

复制代码

BngThea

提示输入目录的时候输入
d:\