一文搞定Python操作文件和目录

简介: 本文首发于公众号“AntDream”,欢迎微信搜索“AntDream”或扫描文章底部二维码关注,和我一起每天进步一点点 文件和目录操作是很常见的功能,这里做个简单的总结,包括注意事项和实际的实现代码,基本日常开发都够用了 目录操作 判断目录或是文件是否存在 os.

本文首发于公众号“AntDream”,欢迎微信搜索“AntDream”或扫描文章底部二维码关注,和我一起每天进步一点点

文件和目录操作是很常见的功能,这里做个简单的总结,包括注意事项和实际的实现代码,基本日常开发都够用了

目录操作

判断目录或是文件是否存在

os.path.exists(path)

判断是否是文件或是目录

# 如果文件或是目录不存在也会返回False
os.path.isfile(path)
os.path.isdir(path)

创建/删除目录

os.mkdir(path)
os.rmdir(path)

得到当前的目录名称

os.path.split(dir_path)[1]

这个方法既简单又实用,比如我们输入一个目录路径:

os.path.split('/usr/projects/project1')[1]
# 输出
project1

# 实际上,上面的代码是分了2步走:
# 第1步:利用os.path.split方法分割路径
os.path.split('/usr/projects/project1')
# 输出
('/usr/projects', 'project1')
# 然后第2步取出第2个结果,也就是当前的目录名字

循环遍历目录

循环遍历目录有2种方式,一种是常规的递归方法,类似下面这样:

def list_file(path):
    for file in os.listdir(path):
        if os.path.isdir(os.path.join(path, file)):
            list_file(os.path.join(path, file))
        print(file)

还有一种更好的方式就是用os.walk方法,类似下面这样:

def list_file_by_walk(dir_path):
    for home, dirs, files in os.walk(dir_path):
        print(home)
        print(dirs)
        # files中是文件列表
        for file_name in files:
            print(file_name)

os.walk方法的几个参数简单介绍一下:

  • home:这个是当前遍历的目录,比如说是 /usr/projects/
  • dirs:是这个目录(home目录)下面的所有文件夹(也就是子目录)
  • files:是这个目录(home目录)下所有的文件

所以从上面的参数中可以看出,os.walk方法会逐一地遍历初始目录下面的所有目录和文件

常见的文件操作

跟上面得到目录名类似的是得到文件的后缀

os.path.splitext(file_path)

这个方法如果输入的是文件的路径,比如:

a = '/usr/projects/project1/test.txt'
os.path.splitext(a)
# 输出
('/usr/projects/project1/test', '.txt')
# 所以跟上面类似,我们可以直接拿到文件的后缀
os.path.splitext(a)[1]

文件的全路径

# 需要文件的目录路径和文件名
os.path.join(dir_name, file_name)

这个方法也很有用,因为我们在打开一个文件的时候都需要知道文件的路径

比如在上面循环遍历目录的例子中,我们可以这样打开编辑文件:

def list_file_by_walk(dir_path):
    for home, dirs, files in os.walk(dir_path):
        print(home)
        print(dirs)
        # files中是文件列表
        for file_name in files:
            # 这个file_name只是一个文件名
            print(file_name)
            # 如果我们需要打开文件进行编辑读写操作,那就需要文件的路径
            # 用os.path.join方法就可以快速得到文件路径
            file_path = os.path.join(home, file_name)

需要注意的是,不要直接用字符串相加的方式来拼接,这样子会有兼容问题,比如windows上可能路径就是错的了

读写文件

打开文件
with open(file, 'r') as f:
    for line in f.readlines():
        print line

大家只要记住上面打开文件的格式就行了,也就是这句:

with open(file_path, 打开的模式) as f
# 我们就拿到了文件对象f,可以对文件进行操作了,比如读写等

原因是文件读写是IO操作,需要及时关闭打开的文件,上述with open() as f的方式会自动帮你关闭文件的,免得自己忘记关

文件的打开模式

文件的打开模式有很多种,比如只读、只写、追加模式等等,具体可以见下图:
文件读写模式--来源于网络

模式有这么多,掌握常见的几种就可以了,其他的等用到了再去了解。需要注意以下几点:

  • 创建文件,只需要打开一个文件即可:
# 只要这样一句就创建了一个新文件,也就是打开以后关闭文件
with open(new_file, 'w') as f:
  • 需要特别注意的是,以w只写模式打开一个文件后会把原文件的内容清除!所以如果在遍历文件的时候,你想同时修改文件就不能简单的以w只写模式打开了。
  • 在遍历文件的时候想同时修改文件怎么办呢?注意这里说的是修改,比如修改某一行。

修改文件的某一部分

这里举例在遍历文件的时候想修改文件,比如以上面遍历目录文件的例子:

def list_file_by_walk(dir_path):
    for home, dirs, files in os.walk(dir_path):
        print(home)
        print(dirs)
        # files中是文件列表
        for file_name in files:
            # 这个file_name只是一个文件名
            print(file_name)
            # 如果我们需要打开文件进行编辑读写操作,那就需要文件的路径
            # 用os.path.join方法就可以快速得到文件路径
            file_path = os.path.join(home, file_name)
            
            # 这个时候我想修改这个文件的内容,要怎么做呢?

需要注意的是,读写模式需要管理文件指针,比较麻烦,写读模式,则会把原先的内容都清除,所以都不适合

一种思路是我先以只读模式打开,然后遍历文件的内容保存起来,比如:

with open(file, 'r') as f:
    for line in f.readlines():
        print line
        # 这里就可以把文件的内容一行行保存起来
        # 找到需要修改的行,修改以后保存

保存以后再以只读模式打开文件,写入即可。

另一种思路是我打开原文件的同时,再打开另一个临时文件用于写入,比如这样:

with open(file, 'r') as f, open(cache_file_path, 'w') as w:
    for line in f.readlines():
        # 中间可以对f文件中的内容进行过滤或是修改
        w.write(line)

# 完了以后需要删除原文件,然后把临时文件的名字修改成原来文件的名字即可

这里就顺道引出了,删除和重命名文件的方法

os.remove(file)
os.rename(cache_file_path, file)

总结

以上就是Python文件和目录操作的常用知识点和方法总结,建议大家收藏起来,以后方便随时查看。

目录
相关文章
|
1月前
|
自然语言处理 数据处理 Python
python操作和解析ppt文件 | python小知识
本文将带你从零开始,了解PPT解析的工具、工作原理以及常用的基本操作,并提供具体的代码示例和必要的说明【10月更文挑战第4天】
310 60
|
1月前
|
安全 Linux 数据安全/隐私保护
python知识点100篇系列(15)-加密python源代码为pyd文件
【10月更文挑战第5天】为了保护Python源码不被查看,可将其编译成二进制文件(Windows下为.pyd,Linux下为.so)。以Python3.8为例,通过Cython工具,先写好Python代码并加入`# cython: language_level=3`指令,安装easycython库后,使用`easycython *.py`命令编译源文件,最终生成.pyd文件供直接导入使用。
python知识点100篇系列(15)-加密python源代码为pyd文件
|
13天前
|
开发者 Python
Python中__init__.py文件的作用
`__init__.py`文件在Python包管理中扮演着重要角色,通过标识目录为包、初始化包、控制导入行为、支持递归包结构以及定义包的命名空间,`__init__.py`文件为组织和管理Python代码提供了强大支持。理解并正确使用 `__init__.py`文件,可以帮助开发者更好地组织代码,提高代码的可维护性和可读性。
15 2
|
1月前
|
Linux 区块链 Python
Python实用记录(十三):python脚本打包exe文件并运行
这篇文章介绍了如何使用PyInstaller将Python脚本打包成可执行文件(exe),并提供了详细的步骤和注意事项。
51 1
Python实用记录(十三):python脚本打包exe文件并运行
|
29天前
|
Java Python
> python知识点100篇系列(19)-使用python下载文件的几种方式
【10月更文挑战第7天】本文介绍了使用Python下载文件的五种方法,包括使用requests、wget、线程池、urllib3和asyncio模块。每种方法适用于不同的场景,如单文件下载、多文件并发下载等,提供了丰富的选择。
|
30天前
|
数据安全/隐私保护 流计算 开发者
python知识点100篇系列(18)-解析m3u8文件的下载视频
【10月更文挑战第6天】m3u8是苹果公司推出的一种视频播放标准,采用UTF-8编码,主要用于记录视频的网络地址。HLS(Http Live Streaming)是苹果公司提出的一种基于HTTP的流媒体传输协议,通过m3u8索引文件按序访问ts文件,实现音视频播放。本文介绍了如何通过浏览器找到m3u8文件,解析m3u8文件获取ts文件地址,下载ts文件并解密(如有必要),最后使用ffmpeg合并ts文件为mp4文件。
|
1月前
|
JSON 数据格式 Python
Python实用记录(十四):python统计某个单词在TXT/JSON文件中出现的次数
这篇文章介绍了一个Python脚本,用于统计TXT或JSON文件中特定单词的出现次数。它包含两个函数,分别处理文本和JSON文件,并通过命令行参数接收文件路径、目标单词和文件格式。文章还提供了代码逻辑的解释和示例用法。
41 0
Python实用记录(十四):python统计某个单词在TXT/JSON文件中出现的次数
|
1月前
|
Python
Python实用记录(四):os模块-去后缀或者改后缀/指定目录下图片或者子目录图片写入txt/csv
本文介绍了如何使用Python的os模块来操作文件,包括更改文件后缀、分割文件路径和后缀、将指定目录下的所有图片写入txt文档,以及将指定目录下所有子目录中的图片写入csv文档,并为每个子目录分配一个标签。
16 1
|
1月前
|
计算机视觉 Python
Python操作PDF文件
Python操作PDF文件
|
1月前
|
Python
Python实用记录(十二):文件夹下所有文件重命名以及根据图片路径保存到新路径下保存
这篇文章介绍了如何使用Python脚本对TTK100_VOC数据集中的JPEGImages文件夹下的图片文件进行批量重命名,并将它们保存到指定的新路径。
33 0