python如何解压缩文件或文件夹

在日常开发或数据处理工作中，我们经常需要对文件或文件夹进行压缩和解压缩操作。Python 提供了强大的内置模块，如 zipfile 和 shutil，可以帮助我们高效地完成这些任务。

本文将介绍如何使用 Python 对文件夹或文件进行压缩和解压缩，内容包括两种常见方式：zipfile 和 shutil，并提供完整代码示例。

压缩文件或文件夹

1. 使用 zipfile 模块压缩文件夹

zipfile 模块是 Python 标准库的一部分，用于创建和读取 .zip 文件。它支持精细控制文件压缩的内容和路径。

以下是一个将文件夹压缩为 .zip 文件的代码示例：

import os
import zipfile

def compress_folder_to_zip(folder_path, zip_path):
    with zipfile.ZipFile(zip_path, 'w', zipfile.ZIP_DEFLATED) as zipf:
        for root, dirs, files in os.walk(folder_path):
            for file in files:
                file_path = os.path.join(root, file)
                # 添加到 ZIP 文件时，保留相对路径
                arcname = os.path.relpath(file_path, folder_path)
                zipf.write(file_path, arcname)
    print(f"压缩完成：{zip_path}")

# 示例用法
compress_folder_to_zip('path/to/folder', 'output.zip')

代码说明：

2. 使用 shutil 模块压缩文件夹

如果你只需要简单快速地将文件夹压缩为 .zip 或 .tar 文件，可以使用 shutil 模块。它支持多种压缩格式，包括 zip、tar、gztar 等。

代码示例如下：

import shutil

def compress_with_shutil(folder_path, output_path, format='zip'):
    # format 支持 "zip", "tar", "gztar", "bztar", "xztar"
    shutil.make_archive(output_path, format, folder_path)
    print(f"压缩完成：{output_path}.{format}")

# 示例用法
compress_with_shutil('path/to/folder', 'output', 'zip')

代码说明：

解压缩文件

1. 使用 zipfile 解压 .zip 文件

使用 zipfile 模块，我们可以解压 .zip 文件到指定目录：

import zipfile

def extract_zip(zip_path, extract_to):
    with zipfile.ZipFile(zip_path, 'r') as zipf:
        zipf.extractall(extract_to)
    print(f"解压完成：{extract_to}")

# 示例用法
extract_zip('output.zip', 'path/to/extract')

代码说明：

2. 使用 shutil 解压文件

如果文件是 .zip 格式，并且不需要复杂操作，可以使用 shutil 快速解压：

import shutil

def extract_with_shutil(zip_path, extract_to):
    shutil.unpack_archive(zip_path, extract_to)
    print(f"解压完成：{extract_to}")

# 示例用法
extract_with_shutil('output.zip', 'path/to/extract')

代码说明：

完整示例：压缩与解压缩结合

以下是一个包含压缩和解压缩功能的完整 Python 脚本：

import os
import zipfile
import shutil

def compress_folder(folder_path, zip_path):
    with zipfile.ZipFile(zip_path, 'w', zipfile.ZIP_DEFLATED) as zipf:
        for root, dirs, files in os.walk(folder_path):
            for file in files:
                file_path = os.path.join(root, file)
                arcname = os.path.relpath(file_path, folder_path)
                zipf.write(file_path, arcname)
    print(f"压缩完成：{zip_path}")

def decompress_folder(zip_path, extract_to):
    with zipfile.ZipFile(zip_path, 'r') as zipf:
        zipf.extractall(extract_to)
    print(f"解压完成：{extract_to}")

# 示例用法
if __name__ == "__main__":
    # 压缩文件夹
    compress_folder('path/to/folder', 'output.zip')
    # 解压 ZIP 文件
    decompress_folder('output.zip', 'path/to/extract')

总结

在这篇文章中，我们学习了如何使用 Python 的 zipfile 和 shutil 模块进行文件夹和文件的压缩与解压缩操作：

无论你是需要对多个文件进行压缩管理，还是希望快速解压一个文件夹，这两种方式都能满足你的需求。

作者：游客520