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

yyyyyyyynnnnn

结构如下：
大文件夹：
        1.txt
        文件夹1
                2.txt
                空文件夹
想要获得所有txt文本的路径，正确输出应该是：
大文件夹\1.txt
大文件夹\文件夹1\2.txt
可是却输出了：
大文件夹\文件夹1\2.txt
大文件夹\文件夹1\空文件夹\1.txt

1. import os
   
2. import os.path
   
3. key=input('请将该脚本放于待查找的文件夹内，请输入关键字：')
   
4. YN=input('请问是否需要打印关键字【%s】在文件中的具体位置（YES/NO）：'%key)
   
5. os.chdir(r'C:\Users\wind\Desktop\test')
   
6. list1=[]
   
7. def listtxt(path):
   
8.     os.chdir(path)
   
9.     for each in os.listdir():
   
10.         (a,b)=os.path.splitext(each)
    
11.         if b=='.txt':
    
12.             list1.append(os.getcwd()+os.sep+each)
    
13.         if os.path.isdir(each):
    
14.             listtxt(each)
    
15. listtxt(r'C:\Users\wind\Desktop\test')

复制代码

最佳答案

月排行榜 / 总排行榜

Twilight6

2021-7-21 08:51:19

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

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

1. import os
   
2. import os.path
   
3. 
   
4. list1=[]
   
5. def listtxt(path):
   
6.     os.chdir(path)
   
7.     for each in os.listdir():
   
8.         (a,b)=os.path.splitext(each)
   
9.         if b == '.txt':
   
10.             list1.append(os.getcwd()+os.sep+each)
    
11.         if os.path.isdir(each):
    
12.             listtxt(each)
    
13.             os.chdir('..')
    
14.             
    
15. listtxt(r'C:\Users\wind\Desktop\test')

复制代码

跳转到最佳答案楼层

Twilight6

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

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

1. import os
   
2. import os.path
   
3. 
   
4. list1=[]
   
5. def listtxt(path):
   
6.     os.chdir(path)
   
7.     for each in os.listdir():
   
8.         (a,b)=os.path.splitext(each)
   
9.         if b == '.txt':
   
10.             list1.append(os.getcwd()+os.sep+each)
    
11.         if os.path.isdir(each):
    
12.             listtxt(each)
    
13.             os.chdir('..')
    
14.             
    
15. listtxt(r'C:\Users\wind\Desktop\test')

复制代码