[已解决]这样为什么无法获得正确文件位置

yyyyyyyynnnnn

结构如下：
大文件夹：
        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')

跳转到最佳答案楼层

Twilight6

因为你递归 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')