DoubleS、Zhao 发表于 2020-11-11 23:34:48

各位大佬帮帮忙,递归的一个问题

import os

def search_file(start_dir, target) :
    os.chdir(start_dir)
   
    for each_file in os.listdir(os.curdir) :
      ext = os.path.splitext(each_file)
      if ext in target :
            vedio_list.append(os.getcwd() + os.sep + each_file + os.linesep) # 使用os.sep是程序更标准
      if os.path.isdir(each_file) :
            search_file(each_file, target) # 递归调用
            os.chdir(os.pardir) # 递归调用后切记返回上一层目录

start_dir = input('请输入待查找的初始目录:')
program_dir = os.getcwd()

target = ['.mp4', '.avi', '.rmvb']
vedio_list = []

search_file(start_dir, target)

f = open(program_dir + os.sep + 'vedioList.txt', 'w')
f.writelines(vedio_list)
f.close()


第六行each_file的目录是当前目录,那在递归调用时each_file通过os.chdir的目录不还是原目录吗?难道会改成each_file里面的一层目录吗?

昨非 发表于 2020-11-11 23:44:47

os.chdir()是将工作目录切换为start的位置
第六行那个不是os.chdir()而是os.curdir,
指代当前目录
这样的话,通过for遍历出来的就是当前目录下的文件内容了

Twilight6 发表于 2020-11-12 00:06:06

本帖最后由 Twilight6 于 2020-11-12 00:08 编辑



每次遍历到文件夹时候,就会进入文件夹

最后文件夹内容遍历完成后,又需要返回出文件夹

否则就会导致遗漏遍历文件夹,照成只打开文件夹不返回的情况

看下这个帖子有助于你的理解不:

[已解决]python递归调用返回上一层目录不理解
https://fishc.com.cn/thread-170223-1-1.html
(出处: 鱼C论坛)
页: [1]
查看完整版本: 各位大佬帮帮忙,递归的一个问题