这样为什么无法获得正确文件位置
结构如下:大文件夹:
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')
因为你递归 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]