9个必知的 Python 操作文件/文件夹方法

Posted 2022-01-04 AI科技大本营

作者 | 欣一

来源 | Python爱好者集中营

近几年随着Python的热度不断上涨，人们渐渐使用这门编程语言来进行一些自动化操作，以节省重复劳动带来的效率低下，那么必定会涉及到对文件系统的操作，包括文件的增、删、改、查等等，今天小编就来介绍一下如何用Python来实现这些功能

输出当前的路径

我们可以通过Python当中的OS库来获取当前文件所在的位置

import os
os.getcwd()

路径的拼接

我们通过os.path.join()方法来实现

os.path.join('output', 'Book1.xlsx')

output

output\\Book1.xlsx

确认某文件夹或者是文件是否存在

我们先来看如何来确认文件夹是否存在，同样也是通过OS模块

os.path.exists('directory_name')

同样的道理，要是我们查看一下某个文件是否存在的时候，可以这么做

os.path.exists('path/file_name')

建立文件夹目录

然后我们来看一下如何新建一个文件夹

os.mkdir("文件夹名称")

当然要是文件夹目录已经提前存在的情况下，上述的代码自然会报错，所以我们通常会先检查一下是否已经存在

if not os.path.exists('文件夹名称'):
    os.mkdir('文件夹名称')

当然我们有时候需要在已建的文件夹下创建子文件夹，例如下图

这个时候要是引用os.mkdir()方法或许会稍显繁琐，这时我们可以通过os.mkdirs()方法来实现

os.makedirs(os.path.join('test_dir', 'level_1', 'level_2', 'level_3'))

罗列出当前目录下所包含的所有文件

代码如下

os.listdir('文件夹名称')

但是有时候我们可能想例如搜寻一下当前目录下的所有文件中以“py”结尾的文件，我们可以通过通配符来实现，代码如下

list(glob(os.path.join('.', '*.py')))

上面提到的glob模块能够快速查找我们想要的目录和文件，它支持*、**、?、[]这四个通配符

移动文件

要是我们想在不同的目录文件夹之下移动文件，该怎么操作呢？这里我们介绍Python当中的shutil模块，假设我们想要将当前目录下的几个csv文件移动到“test_dir”目录文件夹之下，代码如下

import shutil

for file in list(glob(os.path.join('.', '*.csv'))):
    shutil.move(file, 'test_dir')

复制文件

当我们想要复制文件的时候，也可以使用shutil模块，例如我们想要将几个“test_dir”目录文件夹下的csv文件复制到“output”目录文件夹之下，代码如下

shutil.copy(os.path.join('test_dir', 'data.csv'), 'output')

除此之外，我们还能够对粘贴过去的文件重命名，代码如下

shutil.copy(os.path.join('test_dir', 'data.csv'),
            os.path.join('output', 'data_2.csv'))

删除文件

我们来看一下删除文件该如何实现？os.remove()方法可以完成删除文件的功能，

os.remove(os.path.join('output', 'data_2.csv'))

当我们想要删除一整个目录文件夹的时候，可以使用os.rmdir()方法，当然只仅限于是空的目录文件夹

os.rmdir(os.path.join('test_dir', 'level_1', 'level_2', 'level_3'))

而对于目录文件夹下存在文件的，并不是空的目录文件夹，我们还是需要用到shutil模块，代码如下

shutil.rmtree("test_delete")

或者是

shutil.rmtree(os.path.join("test_delete", "test_1_delete"))

创建和解压 压缩包

谈及到操作压缩包相关的内容，就不得不说zipobj这个模块了

• 创建一个压缩包

这里我们要用到的是zipfile模块当中的write()方法

file_lists = list(glob(os.path.join('.', '*.xlsx')))

with zipfile.ZipFile(r"我创建的压缩包.zip", "w") as zipobj:
    for file in file_lists:
        zipobj.write(file)

• 读取压缩包当中的文件信息

通过zipfile模块当中的namelist()方法来实现

with zipfile.ZipFile("我创建的压缩包.zip", "r") as zipobj:
    print(zipobj.namelist())

output

['Book1.xlsx', 'supermarkt_sales.xlsx']

可以看到输出的就是上次我们打包的几个excel文件

• 将压缩包当中的单个文件，解压出来

通过zipfile模块当中的extract()方法来实现

dst = "output"

with zipfile.ZipFile("我创建的压缩包.zip", "r") as zipobj:
    zipobj.extract("Book1.xlsx",dst)

上述代码的用意就是将压缩包当中的“Book1.xlsx”文件解压至“output”目录文件夹下

• 将压缩包中的所有文件，都解压出来

通过zipfile模块当中的extractall()方法来实现，代码如下

dst = "output"

with zipfile.ZipFile("我创建的压缩包.zip", "r") as zipobj:
    zipobj.extractall(dst)

往

期

回

顾

技术

6种常用的绘制地图的方法，码住！

资讯

DeepMind 打造AI游戏系统

技术

20个经典函数说Pandas的数据使用

资讯

机器人Ameca苏醒瞬间逼真到令人...

分享

点收藏

点点赞

点在看