[已解决]零基础学习python，第30讲第3题

gaohhui

编写一个程序，用户输入开始搜索的路径，查找该路径下（包含子文件夹内）所有的视频格式文件（要求查找mp4 rmvb, avi的格式即可），并把创建一个文件（vedioList.txt）存放所有找到的文件的路径：

1. # decoding=gbk
   
2. import os
   
3. 
   
4. 
   
5. def catalog_search(file):
   
6.     list1 = os.listdir(file)
   
7.     for i in list1:
   
8.         if os.path.isdir(i):
   
9.             catalog_search(i)
   
10.         else:
    
11.             # if os.path.splitext(i)[1] == ".dabai":
    
12.                 file_export.writeline(i+"\n")
    
13. 
    
14. 
    
15. catalog = input("请输入待查找的初始目录：")
    
16. list_file = os.listdir(catalog)
    
17. 
    
18. file_export = open("videolist.txt","w")
    
19. 
    
20. for i in list_file:
    
21.     if os.path.isdir(i):
    
22.         catalog_search(i)
    
23.     else:
    
24.         # if os.path.splitext(i)[1] == ".dabai":
    
25.         file_export.writelines(i+"\n")
    
26. 
    
27. file_export.close()

复制代码

我这个代码，只能写入当前文件夹下面的文件，无法进入第二层文件夹，甚至第三层文件夹，请鱼油们帮忙看一下

最佳答案

月排行榜 / 总排行榜

BngThea

2019-3-27 11:18:32

listdir返回的只是一个文件名的字符串列表，你需要将路径进行拼接一下再进行判断是否为一个文件夹

跳转到最佳答案楼层

BngThea

listdir返回的只是一个文件名的字符串列表，你需要将路径进行拼接一下再进行判断是否为一个文件夹

gaohhui

谢谢，我使用os.path.join（）实现了路径拼接