如何使用Python将文件添加到zip文件中？

前言

在日常开发过程中，我们可能需要将多个文件打包成一个压缩包以便于传输或存储。而zip文件是一种常见的压缩文件格式，而Python标准库中的zipfile模块提供了操作zip文件的功能。

本文将介绍如何使用Python将文件添加到zip文件中。

更多Python教程，请阅读：Python 教程

准备工作

在进行文件添加操作之前，需要先了解zipfile模块中重要的几个类和方法：

ZipFile类：用于创建和操作zip文件。
write()方法：向zip文件中写入数据，可以是文件，可以是字符串。
close()方法：关闭zip文件，释放文件句柄。

实现方法

假设我们有一个名为test.txt的文件，我们要将它添加到名为example.zip的zip文件中：

import zipfile

# 创建ZipFile对象，打开example.zip文件
example_zip = zipfile.ZipFile('example.zip', mode='w')

# 将test.txt添加到example.zip文件中
example_zip.write('test.txt')

# 关闭example.zip文件
example_zip.close()

上述代码首先导入了zipfile模块，然后创建了一个zipfile.ZipFile对象example_zip来操作zip文件example.zip。通过调用write()方法，将test.txt添加到example.zip中，最后通过close()方法来关闭文件。

需要注意的是，ZipFile对象的mode参数表示zip文件的打开模式，’w’表示以写入模式打开，如果文件不存在将会创建，如果文件已存在，则会覆盖原有文件内容。还可以使用’a’表示追加模式打开，将数据添加到zip文件末尾。

下面是一个完整的例子，在当前目录下创建example文件夹，并将其中的test1.txt和test2.txt添加到名为example.zip的zip文件中：

import os
import zipfile

# 创建ZipFile对象
example_zip = zipfile.ZipFile('example.zip', mode='w')

# 遍历example文件夹下的所有文件，添加到zip文件中
for dirname, subdirs, files in os.walk('example'):
    example_zip.write(dirname)
    for filename in files:
        example_zip.write(os.path.join(dirname, filename))

# 关闭zip文件
example_zip.close()

在上述代码中，通过使用os.walk()方法遍历了example文件夹下的所有文件，包括子文件夹中的文件，将其添加到名为example.zip的zip文件中。最终得到的压缩包包含有example文件夹和其中的文件。