Python-使用os.walk()实现文件查找

我的电脑中有一个文件查找功能，只要输入需要查找的文件，电脑就会自动搜索某个盘目录下是否存在该文件。我们来使用Python实现该功能吧。

首先导入os模块

import os

这里需要使用os模块的walk()函数。

附os.walk()的使用方法。

os.walk目录遍历

os.walk(top, topdown=True, onerror=None, followlinks=False)

其中： 
- top是要遍历的目录。 
- topdown是代表要从上而下遍历还是从下往上遍历。 
- onerror可以用来设置当便利出现错误的处理函数(该函数接受一个OSError的实例作为参数)，设置为空则不作处理。 
- followlinks表示是否要跟随目录下的链接去继续遍历，要注意的是，os.walk不会记录已经遍历的目录，所以跟随链接遍历的话有可能一直循环调用下去。

os.walk返回的是一个3个元素的元组 (root, dirs, files) ，分别表示遍历的路径名，该路径下的目录列表和该路径下文件列表。注意目录列表和文件列表不是具体路径，需要具体路径(从root开始的路径)的话可以用 os.path.join(root,dir) 和 os.path.join(root,dir) 。
 

使用这个函数就能简单的实现。

a=input('请输入想要查找的磁盘：')+':/'
b=input('请输入想要查找的文件名和后缀：')
for root,dirs,files in os.walk(a,topdown=True):
    if b in files:
        print('找到了')
        print(root)
        break
    else:
        pass

我们测试一下，查找d:磁盘下的b.txt文件目录：

这样就能简单的查找文件了罒ω罒