如何使用Python创建zip文件？

在日常开发和数据分析工作中，我们经常需要打包和压缩一些文件，尤其是对于大型数据集和文件，压缩成一个zip文件后能够极大地减小文件的大小，提高传输和存储效率。Python作为一种广泛应用于各个领域的编程语言，自然也提供了很多方便快捷的实现方法来处理创建zip文件的任务。

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

1. ZipFile模块的基本用法

Python内置的zipfile模块提供了来自文件、流和内存的zip存档的创建，读取，写入和提取的功能。下面是一个简单的示例代码，展示如何使用ZipFile模块从Python代码中创建一个zip文件：

import zipfile

# 创建ZipFile对象
zip_file = zipfile.ZipFile('my_archive.zip', 'w')

# 在zip文件中添加文件
zip_file.write('file1.txt')
zip_file.write('file2.txt')

# 关闭ZipFile对象
zip_file.close()

在上面的代码中，我们首先导入了zipfile模块并创建了一个ZipFile对象。ZipFile对象具有两个必需参数：

file_name（位置参数）：表示在哪个文件（或文件样式）中存储zip归档文件，当你需要创建一个新的zip文件时，可以在对象创建时指定
mode（位置参数）：zip归档文件可以有不同的模式，它们分别代表不同的功能。我们可以在对象创建时指定其中一个，常用的有'w'和'x'。w代表zip存档文件已存在并且可以被写入，而x代表新建的zip存档文件。

在创建ZipFile对象后，我们就可以使用write()方法向zip文件中添加文件。我们可以连续多次调用该方法，以添加更多的文件。这个方法只接受位置参数，表示要添加到zip文件中的文件路径。还可以使用关键字参数，用于设置保存在zip文件中的文件的属性。例如，我们可以设置文件名和压缩格式等。

最后，我们必须调用close()方法来关闭ZipFile对象。如果在手动调用该方法之前没有关闭对象，则可能会出现未写入的存档内容。

2. 自动添加文件夹

当我们需要向zip文件中添加来自文件夹或者文件夹中的子文件的时候，更换每个文件和子文件的名称会变得非常繁琐。然而，Python中的ZipFile模块提供了一种更好的方法，即使用ZipFile类中的write()方法加上arcname关键字参数来实现。

import zipfile
import os

# 创建一个zip文件
zip_file = zipfile.ZipFile("my_archive.zip", mode="w")

# 添加文件夹到zip文件
dir_path = "my_folder"
for root, dirs, files in os.walk(dir_path):
    for file in files:
        file_path = os.path.join(root, file)
        zip_file.write(file_path, arcname=os.path.relpath(file_path, dir_path))

# 关闭zip文件
zip_file.close()

在上述示例代码中，我们首先为zip文件创建一个ZipFile对象，并使用Python内置的os模块遍历文件夹中的每个文件。使用os.path方法可以对文件和文件夹进行操作，例如以字符串形式返回文件或目录的绝对路径和基本名称。之后，我们使用write()方法向zip文件中添加每个文件，指定arcname参数为文件相对于初始文件夹的路径。这样，无论在Windows系统中还是在Unix-like系统中，添加到zip文件的文件将会拥有正确的文件夹。

3.创建压缩文件时添加注释

当我们创建zip文件时，可以选择添加注释，以帮助用户更好地理解和使用该文件。Python的ZipFile模块提供了ZipInfo类，用于描述zip中的文件，其中的一个属性是文件的注释。我们可以使用该类来为zip文件添加注释，示例代码如下：

import zipfile

# 创建ZipFile对象
zip_file = zipfile.ZipFile('my_archive.zip', 'w')

# 为zip文件添加注释
zip_comment = "This is a sample zip file created with Python."
zip_file.comment = zip_comment.encode('utf-8')

# 在zip文件中添加文件
zip_file.write('file1.txt')
zip_file.write('file2.txt')

# 关闭ZipFile对象
zip_file.close()

在上述示例代码中，我们使用了Python的zip_comment来指定zip文件的注释。我们使用encode()方法将字符串转换为bytes类型，因为ZipFile类的comment属性是bytes类型。最后，我们将ZipFile对象关闭。

4. 将zip文件解压缩

在实际应用中，我们有时需要从zip文件中提取数据，或将整个zip文件解压缩到文件夹中。Python的ZipFile模块可以帮助我们解压文件。在下面的示例代码中，我们将zip文件解压缩到当前文件夹中：

import zipfile

# 打开zip文件
my_archive = zipfile.ZipFile("my_archive.zip")

# 提取zip文件中的所有文件
my_archive.extractall()

# 关闭zip文件
my_archive.close()

在上述示例代码中，我们首先使用ZipFile类打开zip文件，然后使用extractall()方法将zip文件中的所有文件提取到当前目录（即Python脚本正在运行的目录）中。

在extractall()方法中也可以指定提取到指定目录中，例如：

import zipfile

# 打开zip文件
my_archive = zipfile.ZipFile("my_archive.zip")

# 提取zip文件中的所有文件到指定目录
my_archive.extractall("extracted_files")

# 关闭zip文件
my_archive.close()

在上述示例代码中，我们将zip文件解压缩到了extracted_files文件夹中。