yyyyyyyynnnnn 发表于 2021-7-20 20:32:06

这样为什么无法获得正确文件位置

结构如下:
大文件夹:
        1.txt
        文件夹1
                2.txt
                空文件夹
想要获得所有txt文本的路径,正确输出应该是:
大文件夹\1.txt
大文件夹\文件夹1\2.txt
可是却输出了:
大文件夹\文件夹1\2.txt
大文件夹\文件夹1\空文件夹\1.txt
import os
import os.path
key=input('请将该脚本放于待查找的文件夹内,请输入关键字:')
YN=input('请问是否需要打印关键字【%s】在文件中的具体位置(YES/NO):'%key)
os.chdir(r'C:\Users\wind\Desktop\test')
list1=[]
def listtxt(path):
    os.chdir(path)
    for each in os.listdir():
      (a,b)=os.path.splitext(each)
      if b=='.txt':
            list1.append(os.getcwd()+os.sep+each)
      if os.path.isdir(each):
            listtxt(each)
listtxt(r'C:\Users\wind\Desktop\test')

Twilight6 发表于 2021-7-21 08:51:19


因为你递归 os.chdir 更改了工作目录,忘记返回上一层目录了,在递归函数下面加个 os.chdir('..') 即可

你代码前面的 input 没什么用,因为你想找的是 txt 文件,在你代码上稍微修改了下,参考代码:

import os
import os.path

list1=[]
def listtxt(path):
    os.chdir(path)
    for each in os.listdir():
      (a,b)=os.path.splitext(each)
      if b == '.txt':
            list1.append(os.getcwd()+os.sep+each)
      if os.path.isdir(each):
            listtxt(each)
            os.chdir('..')
            
listtxt(r'C:\Users\wind\Desktop\test')
页: [1]
查看完整版本: 这样为什么无法获得正确文件位置