【Python】9个必知的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.makedirs(os.path.join('test_dir', 'level_1', 'level_2', 'level_3'))

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

代码如下

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

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

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

移动文件

import shutil

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

复制文件

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.path.join('test_dir', 'level_1', 'level_2', 'level_3'))

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)