祝建宇 发表于 2020-8-21 20:52:35

文件系统第三题

import os

vedio_list = []

def search_file(start_dir):
    os.chdir(start_dir)
    for each_file in os.listdir(os.curdir):
      if os.path.isfile(each_file):
            file_ext = os.path.splitext(each_file)
            if file_ext in ['.mp4', '.rmvb', '.avi']:
                vedio_list.append(os.getcwd() + os.sep + each_file + os.linesep)
      if os.path.isdir(each_file):
            search_file(each_file)# 递归调用
            os.chdir(os.pardir)# 递归调用后切记返回上一层目录
    return vedio_list

start_dir = input('请输入待查找的初始目录：')
vedio_list = search_file(start_dir)
f = open(os.getcwd() + os.sep + 'VedioList.txt', 'w')
f.writelines(vedio_list)
f.close()

红线地方为什么要修改文件目录？

Twilight6 发表于 2020-8-21 20:58:19

本帖最后由 Twilight6 于 2020-8-21 21:00 编辑



因为你下面的 os.listdir(os.curdir) 是列举当前文件夹下的文件，返回文件名字符串列表

而通过直接修改了代码当前工作目录，就可以配合 os.listdir(os.curdir) 直接读取列举修改后工作目录的文件

你也可以使用绝对路径，但是比较麻烦，需要多次通过 split 切割字符串路径，来达到这个目的

1q23w31 发表于 2020-8-21 20:58:34

为了在你输入的目录里进行操作，python默认只能对程序所在目录操作，所以需要改变目录
页: [1]
查看完整版本: 文件系统第三题