Python 如何查找特定类型文件


Posted in Python onAugust 17, 2020

写在之前

今天的文章是介绍如何用 Python 去定位特定类型的文件,会讲到用字符串匹配文件名定位特定文件以及顺带介绍一下遍历目录树的函数,通过今天的这一部分以及之前文章讲到的文件获取属性的操作,可以做很多有意思的事情。

定位特定文件

定位特定的文件,可以使用 fnmatch 以及 glob 这两个标准库,我们下面来分别看一下。

1. 使用 fnmatch 标准库

一般的话我们想要查找特定类型的文件,可以通过字符串的前缀匹配和后缀匹配来查找,具体实例如下所示:

>>> import os
>>> [txt for txt in os.listdir('.') if txt.endswith('.txt')]
['b.txt', 'a.txt']

其实大部分的情况下这种用字符串匹配的方法来查找文件就足够可以满足需要,如果有些情况下需要更加灵活一点的字符串匹配,就可以使用 fnmatch 库,这是一个专门用来进行文件名匹配的库,支持使用通配符来进行文字匹配(比如'*','?'等)。

fnmatch 这个库很简单,只有 4 个函数:fnmatch,fnmatchcase,filter,translate:

  • fnmatch:判断文件名是否符合特定的模式;
  • fnmatchcase:判断文件名是否符合特定的模式(不区分大小写);
  • filter:返回输入列表中,符合特定模式的文件名列表;
  • translate:将通配符模式转换成正则表达式。

上面 4 个最常用是它的同名函数 fnmatch ,我的当前目录下有 4 个文件,分别是:a.txt,b.txt,c.py,test.py,下面我就用 fnmatch 函数来对当前目录下的这 4 个文件进行匹配性测试,具体请看下面代码:

>>> import os
>>> import fnmatch
>>> os.listdir('.')
['test.py', 'c.py', 'b.txt', 'a.txt']
>>> [txt for txt in os.listdir('.') if fnmatch.fnmatch(txt, '*.txt')]
['b.txt', 'a.txt']
>>> [file for file in os.listdir('.') if fnmatch.fnmatch(file, '[a-c]*')]
['c.py', 'b.txt', 'a.txt']
>>> [file for file in os.listdir('.') if fnmatch.fnmatch(file, '[!a-c]*')]
['test.py']

fnmatchcase 函数与 fnmatch 函数几乎一样,只是在匹配的时候会忽略大小写字母;filter 函数与 fnmatch 比较类似,区别在于fnmatch 一次只对一个文件名匹配,而 filter 一次可以对多个文件名进行匹配判断,下面我们就再来看一下 filter 的用法:

>>> import os
>>> import fnmatch
>>> file = os.listdir('.')
>>> file
['test.py', 'c.py', 'b.txt', 'a.txt']
>>> fnmatch.filter(file, '[a-c]*')
['c.py', 'b.txt', 'a.txt']
>>> fnmatch.filter(file, '[!a-c]*')
['test.py']

2.使用 glob 定位

我们在上面介绍的获取特定类型的文件列表,首先是通过 os.listdir 获取全部的文件列表,然后通过字符串匹配或者使用 fnmatch 进行文件名模式匹配来定位,习惯了 Python 的简洁优雅,这多出来的一步总觉得哪里怪怪的,好在我现在知道了 glob。

glob 的作用相当于 os.listdir + fnmatch 的组合,使用 glob 以后,就不需要用 os.listdir 获取文件列表,直接通过模式匹配就可以解决了,具体操作如下所示:

>>> import glob
>>> glob.glob('*.txt')
['b.txt', 'a.txt']
>>> glob.glob('[a-c]*')
['c.py', 'b.txt', 'a.txt']
>>> glob.glob('[!a-c]*')
['test.py']

可以看到,Python 真的非常的灵活,光是定位目录下特定的文件类型我就已经介绍了 3 种方式,一般情况下就用字符串匹配就可以解决,如果需要更加灵活一点的的,可以使用 fnmatch 和 glob。

遍历目录树

再上一节中我们都是查找某个目录下的文件然后通过匹配去定位自己需要的文件类型。但是在实际的应用过程中我们更可能遇到的是某个目录及其子目录下的所有文件。比如查找某个目录及其子目录下所有的 txt 文件等等,对于这类的需求,我们可以使用 os 模块下的「walk」函数。walk 函数遍历某个目录及其子目录,对于每一个目录,walk 返回一个三元组,依次是「当前目录」,「当前目录下的子目录」,「当前目录下的文件列表」。

下面我们来演示一下 os.walk 函数的用法,做一个小的练习,来遍历一下 /root/rocky0429 目录及其子目录下的所有 txt 和 图片文件:

import os
import fnmatch

search = ['*.txt', '*.jpg', '*.jpeg']
res = []

for root, dirnames, filenames in os.walk(os.path.expanduser('~/rocky0429')):
  for extension in search:
    for filename in fnmatch.filter(filenames, extension):
      res.append(os.path.join(root,filename))

    print(res)

以上就是Python 如何查找特定类型文件的详细内容,更多关于python 查找文件的资料请关注三水点靠木其它相关文章!

Python 相关文章推荐
Python安装使用命令行交互模块pexpect的基础教程
May 12 Python
python变量不能以数字打头详解
Jul 06 Python
python编写微信远程控制电脑的程序
Jan 05 Python
Python cookbook(数据结构与算法)找到最大或最小的N个元素实现方法示例
Feb 13 Python
深入分析python数据挖掘 Json结构分析
Apr 21 Python
Python实现重建二叉树的三种方法详解
Jun 23 Python
Python爬虫使用脚本登录Github并查看信息
Jul 16 Python
Python 生成一个从0到n个数字的列表4种方法小结
Nov 28 Python
python读取与处理netcdf数据方式
Feb 14 Python
Python接口自动化判断元素原理解析
Feb 24 Python
python实现将字符串中的数字提取出来然后求和
Apr 02 Python
pip install命令安装扩展库整理
Mar 02 Python
Python 通过爬虫实现GitHub网页的模拟登录的示例代码
Aug 17 #Python
无需压缩软件,用python帮你操作压缩包
Aug 17 #Python
用Python制作mini翻译器的实现示例
Aug 17 #Python
PyCharm vs VSCode,作为python开发者,你更倾向哪种IDE呢?
Aug 17 #Python
pandas apply多线程实现代码
Aug 17 #Python
简述python Scrapy框架
Aug 17 #Python
python使用多线程查询数据库的实现示例
Aug 17 #Python
You might like
session在php5.3中的变化 session_is_registered() is deprecated in
2013/11/12 PHP
php中Y2K38的漏洞解决方法实例分析
2014/09/22 PHP
PHP封装的XML简单操作类完整实例
2017/11/13 PHP
List Installed Hot Fixes
2007/06/12 Javascript
JS 常用校验函数
2009/03/26 Javascript
兼容多浏览器的iframe自适应高度(ie8 、谷歌浏览器4.0和 firefox3.5.3)
2009/11/04 Javascript
iframe窗口高度自适应的又一个巧妙实现思路
2014/04/04 Javascript
js实现获取焦点后光标在字符串后
2014/09/17 Javascript
jQuery中removeData()方法用法实例
2014/12/27 Javascript
javascript使用shift+click实现选择和反选checkbox的方法
2015/05/04 Javascript
javascript自定义右键弹出菜单实现方法
2015/05/25 Javascript
jquery实现鼠标点击后展开列表内容的导航栏效果
2015/09/14 Javascript
jQuery实现删除li节点的方法
2016/12/06 Javascript
浅析上传头像示例及其注意事项
2016/12/14 Javascript
从零开始学习搭建React脚手架项目
2018/08/23 Javascript
深入分析element ScrollBar滚动组件源码
2019/01/22 Javascript
vue实现侧边栏导航效果
2019/10/21 Javascript
vue 实现购物车总价计算
2019/11/06 Javascript
基于JS实现操作成功之后自动跳转页面
2020/09/25 Javascript
微信小程序实现页面左右滑动
2020/11/16 Javascript
[02:32]DOTA2英雄基础教程 美杜莎
2014/01/07 DOTA
Python中的ceil()方法使用教程
2015/05/14 Python
Python使用PIL库实现验证码图片的方法
2016/03/11 Python
Python操作mysql数据库实现增删查改功能的方法
2018/01/15 Python
Python图像处理之图像的缩放、旋转与翻转实现方法示例
2019/01/04 Python
在python中实现调用可执行文件.exe的3种方法
2019/07/07 Python
Python time库基本使用方法分析
2019/12/13 Python
Python描述符descriptor使用原理解析
2020/03/21 Python
python如何变换环境
2020/07/21 Python
一个大学生十年的职业规划
2014/01/17 职场文书
房务中心文员岗位职责
2014/04/16 职场文书
2014年祖国生日寄语
2014/09/19 职场文书
开业庆典活动策划方案
2014/09/21 职场文书
走近毛泽东观后感
2015/06/04 职场文书
领导离职感言
2015/08/03 职场文书
swagger如何返回map字段注释
2021/07/03 Java/Android