Python如何压缩文件？Zipfile模块教程

python压缩文件的核心是zipfile模块，它提供了创建、读取、写入和提取zip文件的功能。1. 创建zip文件：使用zipfile类配合'w'模式，将指定文件列表写入新压缩包。2. 添加文件到现有zip：通过'a'模式追加文件而不覆盖原文件。3. 提取zip文件：用'r'模式结合extractall()方法将内容解压到指定目录。4. 读取zip内容：调用namelist()方法获取压缩包内所有文件列表。5. 指定压缩级别：通过compresslevel参数设置压缩比，范围0-9，9为最高但较慢。6. 处理大型文件：使用分块读取配合writestr()方法避免内存过载。7. 排除特定文件：遍历目录时检查文件路径是否匹配排除模式。8. 解决中文乱码：创建zipfile时指定encoding参数如utf-8或gbk。9. 加密zip文件：使用setpassword()设置密码，但zipcrypto加密较弱，敏感数据建议使用第三方库如pyzipper实现aes加密。

Python压缩文件，核心在于使用zipfile模块。它能让你轻松创建、读取、写入和提取ZIP文件，就像操作普通文件一样。

解决方案

zipfile模块是Python处理ZIP文件的瑞士军刀。以下是一些关键用法：

立即学习“Python免费学习笔记（深入）”；

1. 创建ZIP文件：

   import zipfile
   
   def create_zip(filename, file_list):
       with zipfile.ZipFile(filename, 'w') as zip_file:
           for file in file_list:
               zip_file.write(file)
   
   # 例子
   files_to_compress = ['file1.txt', 'file2.txt', 'image.png']
   create_zip('my_archive.zip', files_to_compress)

   登录后复制

   这段代码创建了一个名为my_archive.zip的压缩包，并将file1.txt、file2.txt和image.png添加到其中。注意，'w'模式表示写入，如果文件已存在会被覆盖。

   

2. 添加文件到现有ZIP：

   import zipfile
   
   def add_to_zip(zip_filename, file_to_add):
       with zipfile.ZipFile(zip_filename, 'a') as zip_file:
           zip_file.write(file_to_add)
   
   # 例子
   add_to_zip('my_archive.zip', 'new_file.txt')

   登录后复制

   这里，'a'模式表示追加，将new_file.txt添加到已存在的my_archive.zip中。

3. 提取ZIP文件：

   import zipfile
   
   def extract_zip(zip_filename, extract_path):
       with zipfile.ZipFile(zip_filename, 'r') as zip_file:
           zip_file.extractall(extract_path)
   
   # 例子
   extract_zip('my_archive.zip', 'extracted_files')

   登录后复制

   这段代码会将my_archive.zip中的所有文件提取到extracted_files目录。确保目录存在，否则会报错。

4. 读取ZIP文件内容：

   import zipfile
   
   def list_zip_contents(zip_filename):
       with zipfile.ZipFile(zip_filename, 'r') as zip_file:
           print(zip_file.namelist())
   
   # 例子
   list_zip_contents('my_archive.zip')

   登录后复制

   namelist()方法返回ZIP文件中所有文件的列表。

   

   文心大模型

   百度飞桨-文心大模型 ERNIE 3.0 文本理解与创作

   56

   查看详情

5. 指定压缩级别：

   import zipfile
   
   def create_zip_with_compression(filename, file_list, compress_level=zipfile.ZIP_DEFLATED):
       with zipfile.ZipFile(filename, 'w', compresslevel=compress_level) as zip_file:
           for file in file_list:
               zip_file.write(file)
   
   # 例子
   files_to_compress = ['large_file.txt']
   create_zip_with_compression('compressed.zip', files_to_compress, compress_level=9) # 9是最高压缩级别

   登录后复制

   compresslevel参数允许你控制压缩级别。zipfile.ZIP_DEFLATED是常用的压缩方法，级别范围是0-9，9是最高压缩比，但速度较慢。

如何处理大型文件压缩？

处理大型文件时，一次性加载到内存可能导致问题。可以使用ZipFile.write()的arcname参数和分块读取的方式。

import zipfile
import os

def compress_large_file(zip_filename, filename, chunk_size=4096):
    with zipfile.ZipFile(zip_filename, 'w', zipfile.ZIP_DEFLATED) as zf:
        with open(filename, 'rb') as f:
            # 使用 arcname 指定压缩包内的文件名
            zf.writestr(filename, f.read())

这种方法避免了将整个文件加载到内存中，而是分块读取并写入ZIP文件。

压缩时如何排除特定文件或目录？

有时你可能需要排除某些文件或目录。可以通过在添加文件之前进行检查来实现。

import zipfile
import os

def create_zip_with_exclusion(zip_filename, source_dir, exclude_patterns):
    with zipfile.ZipFile(zip_filename, 'w', zipfile.ZIP_DEFLATED) as zip_file:
        for root, _, files in os.walk(source_dir):
            for file in files:
                filepath = os.path.join(root, file)
                # 检查是否匹配任何排除模式
                if any(pattern in filepath for pattern in exclude_patterns):
                    continue  # 跳过该文件
                zip_file.write(filepath, os.path.relpath(filepath, source_dir))

# 例子
source_directory = 'my_project'
exclude_list = ['.git', '__pycache__', '*.log'] # 排除.git目录、__pycache__目录和所有.log文件
create_zip_with_exclusion('project_archive.zip', source_directory, exclude_list)

这个例子展示了如何排除.git目录、__pycache__目录和所有.log文件。os.walk()遍历目录树，然后检查每个文件是否匹配排除模式。

压缩文件时出现中文乱码怎么办？

在处理包含中文的文件名时，可能会遇到乱码问题。这通常与编码有关。可以尝试在创建ZipFile对象时指定encoding参数。

import zipfile

def create_zip_with_encoding(filename, file_list, encoding='utf-8'):
    with zipfile.ZipFile(filename, 'w', encoding=encoding) as zip_file:
        for file in file_list:
            zip_file.write(file)

# 例子
files_to_compress = ['中文文件.txt']
create_zip_with_encoding('chinese_archive.zip', files_to_compress)

通过指定encoding='utf-8'，可以确保中文文件名正确编码。如果utf-8仍然有问题，可以尝试其他编码，如gbk。

如何加密ZIP文件？

zipfile模块本身并不直接支持强加密（如AES）。但可以使用密码保护（ZipCrypto），虽然这种加密方式相对较弱。

import zipfile

def create_encrypted_zip(filename, file_list, password):
    with zipfile.ZipFile(filename, 'w', zipfile.ZIP_DEFLATED) as zip_file:
        for file in file_list:
            zip_file.setpassword(password.encode('utf-8'))
            zip_file.write(file)

# 例子
files_to_compress = ['sensitive_data.txt']
password = 'my_secret_password'
create_encrypted_zip('encrypted_archive.zip', files_to_compress, password)

注意，setpassword()方法需要一个字节串，所以需要将密码编码为utf-8。

要提取加密的ZIP文件，也需要在ZipFile对象上调用setpassword()。

import zipfile

def extract_encrypted_zip(zip_filename, extract_path, password):
    with zipfile.ZipFile(zip_filename, 'r') as zip_file:
        zip_file.setpassword(password.encode('utf-8'))
        zip_file.extractall(extract_path)

# 例子
password = 'my_secret_password'
extract_encrypted_zip('encrypted_archive.zip', 'extracted', password)

再次强调，ZipCrypto加密相对较弱，对于高度敏感的数据，应考虑使用更强的加密方法，例如使用第三方库pyzipper，它支持更安全的AES加密。

以上就是Python如何压缩文件？Zipfile模块教程的详细内容，更多请关注php中文网其它相关文章！