python查找子目录文件名问题

[jiyiliuniany]

原题为：编写一个search(s)的函数，能在当前目录以及当前目录的所有子目录下查找文件名包含指定字符串的文件，并打印出完整路径。

以下为我的代码

import os
def search(s):
	for x in os.listdir('.'):
		if os.path.isfile(x) and (os.path.splitext(x)[0].find(s) != -1):
			print(x)
		elif os.path.isdir(x):
			newpath = os.path.join('/home/jiiyiliunian/Desktop', str(x))
			os.chdir(newpath)
			search(s)
		else:
			print("error")

search("test")

运行结果如下：


如果注释掉elif和else部分的代码，运行正常


初学python的小白实在找不到问题了。。请教前辈们，代码哪里出错了？谢谢

[extend]

有逻辑漏洞，如果一个文件即不满足if，又不满足elif，那就执行print（"error"），然后，就没有然后了，函数就退出了。

• 打赏
• 举报

回复 赞

[extend]

import os
def search(s,path):
    print(os.listdir(path))
    for x in os.listdir(path):
        #print(x)
        if os.path.isfile(x) and (os.path.splitext(x)[0].find(s) != -1):
            print(x)
        
        elif os.path.isdir(x):
            newpath = os.path.join(path, str(x))
            print(newpath)
            #os.chdir(newpath)
            search(s,newpath)
        else:
            print("error")
            #return
        
search("test",".")

改了一下，可用，你试试。 不用os.chdir(newpath)这种改变当前目录的方式。 递归，你要用传目录的方式。

• 打赏
• 举报

回复 赞