python关于os.walk函数查找windows文件方式
作者:愿此后再无WA
这篇文章主要介绍了python关于os.walk函数查找windows文件方式,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教
前言
OS模块是与操作系统交互的模块。这其中一个非常实际且常用的功能莫过于帮助我们查找需要的文件。
由于前面我发布过一篇比较详细的OS模块讲解文章,这里我就不再多述。
https://www.jb51.net/python/2959318dw.htm
直接上代码!!
查找文件
import os # 文件查找 def file_search(): # os.walk的三个返回值分别是 路径 目录列表 文件列表 它会不断的进入每个目录获取信息 for dir_path, listdir, files in os.walk(path): for file in files: info = dir_path + os.sep + file print(info) # 如果名字符合则用列表储存起来 # 可以把 in 换成 == 吗? 不建议,因为可能传入的文件名不带格式的,但是实际文件是有格式后缀的,如XX.txt if target_name in file: result.append(info) print() output() # 文件输出 def output(): if result: print("文件出现在如下位置:") for i in result: print(i) else: print("未查询到文件!") if __name__ == '__main__': path = input("请输入查找路径(默认在C盘下寻找,可直接回车):") path = 'C:\\' if not path else path # 若有输出路径则使用给出的,否则使用默认的C盘 # 如果给出的路径存在才进行查找 if os.path.exists(path): target_name = input("请输入需要查找的文件名:") result = [] file_search() else: print("您输入的路径不存在!")
其他需求
查找目录
我们可以使用os.path模块下的basename方法,通过路径来截取目录名,再看是否匹配。
那样的话更方便,还不用看文件列表了。
def file_search(): # os.walk的三个返回值分别是 路径 目录列表 文件列表 它会不断的进入每个目录获取信息 for dir_path, listdir, files in os.walk(path): if os.path.basename(dir_path) == target_name: result.append(dir_path) print() output()
提升速度
print方法会消耗不少时间,适当的时候我们可以不查看当前的路径信息,直接等结果就好了。
另外,如果你知道有多少个文件的话可以设置一个终止条件,当找到的文件个数满足条件时,直接终止程序。
根据文件格式查找文件
如果知道文件格式,可以根据目标文件的后缀匹配。
如:我想在某个找所有MP3文件或其他,就可以这么做
# 文件查找 def file_search(): # os.walk的三个返回值分别是 路径 目录列表 文件列表 它会不断的进入每个目录获取信息 for dir_path, listdir, files in os.walk(path): for file in files: info = dir_path + os.sep + file # endswith 以···结尾 如果要查MP3文件,那么 target_name = MP3 if file.endswith(target_name): result.append(info) print() output()
模糊查找
如果你知道某个比较重要的关键词可以用str1 in str2语句判断是否是符合条件的文件;如果你知道更多特征可以使用正则表达式查找。
比如我想找这几个文件,但我只知道Redis 和 ()两个关键字符串,那么应该这么写。
# 文件查找 def file_search(): # os.walk的三个返回值分别是 路径 目录列表 文件列表 它会不断的进入每个目录获取信息 for dir_path, listdir, files in os.walk(path): for file in files: info = dir_path + os.sep + file # target_name = '.*[Rr]edis.*[((].[))]' ret = re.match(target_name, file) if ret: result.append(info) print() output()
其中:
.表示任意单个字符,
*表示*前面的那个字符能够出现任意多次。
.*表示任意字符出现任意多次。
我不知道redis的r是大写还是小写所以用[Rr]表示,[Rr]表示单个字符可选R也可选r。
不知道括号时中文还是英文的用[((].[))]表示。
总结
以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。