如何使用 Python 从 zip 文件中提取所有 .txt 文件？

---

多个文件可以使用ZIP压缩包一起压缩和存储，这在数据处理和文件管理领域很常见。Python 是一种灵活而强大的语言，提供了许多模块来无缝处理ZIP文件。从ZIP压缩包中提取特定文件，例如所有.txt文件，是一个常见的任务。本文将深入探讨使用Python从ZIP包中提取所有.txt文件的过程。我们将逐步讲解原理，并提供一些现实世界的代码示例来说明这个过程。

从ZIP压缩包中提取所有文件

让我们先学习如何从ZIP包中提取所有文件。此示例将为我们将提取和过滤.txt文件的其他示例奠定基础。关键在于：

示例

在这里，我们定义了`extract_all_files`方法，它接受ZIP包的路径和应将其提取到的文件夹作为参数。我们可以使用`zipfile`打开ZIP压缩包。使用以读取模式('r')打开的`ZipFile()`中的`extractall()`函数将所有文件提取到指定的目标文件夹。

import zipfile

def extract_all_files(zip_file_path, extract_to):
   with zipfile.ZipFile(zip_file_path, 'r') as zip_ref:
      zip_ref.extractall(extract_to)

# Example usage
zip_file_path = 'my_archive.zip'
extract_to = 'destination_folder'
extract_all_files(zip_file_path, extract_to)

提取特定文件类型

现在我们已经了解了如何从ZIP包中提取所有文件，我们可以专注于提取特定文件类型，特别是.txt文件。我们可以通过迭代ZIP包中的文件列表并仅选择扩展名为.txt的文件来实现这一点。请查看代码：

示例

在这段代码中，我们循环遍历`zip_ref.infolist()`返回的文件详细信息列表。我们使用`endswith()`函数来检查每个文件的名称是否以.txt结尾。如果是，我们使用`zip_ref.extract()`函数将该特定文件提取到指定的目的地文件夹。

import zipfile

def extract_txt_files(zip_file_path, extract_to):
   with zipfile.ZipFile(zip_file_path, 'r') as zip_ref:
      for file_info in zip_ref.infolist():
         if file_info.filename.endswith('.txt'):
            zip_ref.extract(file_info, extract_to)

# Example usage
zip_file_path = 'my_archive.zip'
extract_to = 'destination_folder'
extract_txt_files(zip_file_path, extract_to)

将文件提取到特定的目录结构

在许多情况下，在从ZIP压缩包中提取数据时保持目录结构至关重要。例如，如果ZIP压缩包包含文件夹，我们可能希望在提取时保留这些文件夹。让我们看看如何做到这一点：

示例

在这个例子中，我们使用`os.path.join()`构建目标文件路径，以确保提取的文件存储在正确的目录结构中。`file_info.filename`提供了文件在ZIP压缩包中的相对路径，`os.path.join()`函数将此路径与提取目录连接起来，从而生成最终文件路径。

import zipfile
import os

def extract_txt_files_with_structure(zip_file_path, extract_to):
   with zipfile.ZipFile(zip_file_path, 'r') as zip_ref:
      for file_info in zip_ref.infolist():
         if file_info.filename.endswith('.txt'):
            file_path = os.path.join(extract_to, file_info.filename)
            zip_ref.extract(file_info, file_path)

# Example usage
zip_file_path = 'my_archive.zip'
extract_to = 'destination_folder'
extract_txt_files_with_structure(zip_file_path, extract_to)

提取具有前缀的文件

有时，我们可能希望提取具有特定前缀的文件，而不管其扩展名如何。例如，我们可能希望提取所有名称中包含“data_”字样的文件。让我们研究一下如何做到这一点：

示例

在这段代码片段中，我们使用`startswith()`函数来检查每个文件的名称是否以所需的前缀开头。如果是，我们在将其提取到所需位置时保留该特定文件的相对路径结构。

import zipfile
import os

def extract_files_with_prefix(zip_file_path, extract_to, prefix):
   with zipfile.ZipFile(zip_file_path, 'r') as zip_ref:
      for file_info in zip_ref.infolist():
         if file_info.filename.startswith(prefix):
            file_path = os.path.join(extract_to, file_info.filename)
            zip_ref.extract(file_info, file_path)

# Example usage
zip_file_path = 'my_archive.zip'
extract_to = 'destination_folder'
prefix = 'data_'
extract_files_with_prefix(zip_file_path, extract_to, prefix)

使用自定义提取函数提取文件

如果特殊情况需要我们执行更复杂的提取怎么办？这可以通过使用自定义提取函数来完成。看看如何实践：

示例

在此示例中，定义了一个名为`custom_extraction_func()`的自定义提取函数。它接受一个`file_info`对象作为输入，并根据某些条件返回True或False。在本例中，自定义条件是.txt扩展名和大于1024字节的文件大小。

`custom_extract()`函数接受ZIP压缩包路径、目标文件夹和具体的提取函数作为参数。然后，每个ZIP压缩包中的文件都会调用特定的提取函数，只有当函数返回True时才会进行提取。

import zipfile
import os

def custom_extraction_func(file_info):
   # Your custom condition here
   return file_info.filename.endswith('.txt') and file_info.file_size > 1024

def custom_extract(zip_file_path, extract_to, extraction_func):
   with zipfile.ZipFile(zip_file_path, 'r') as zip_ref:
      for file_info in zip_ref.infolist():
         if extraction_func(file_info):
            file_path = os.path.join(extract_to, file_info.filename)
            zip_ref.extract(file_info, file_path)

# Example usage with the custom_extraction_func
zip_file_path = 'my_archive.zip'
extract_to = 'destination_folder'
custom_extract(zip_file_path, extract_to, custom_extraction_func)

结论

在这篇深入的文章中，我们探讨了如何使用Python灵活的`zipfile`模块从ZIP包中提取所有.txt文件。首先，我们理解了`zipfile`模块和ZIP文件的基础知识。

然后，我们提供了一些现实世界的代码示例，涵盖了各种情况，包括提取所有文件、提取特定文件类型、保持目录结构、提取具有前缀的文件以及创建自定义提取函数。

现在，借助您新获得的专业知识，您可以轻松地处理Python中的ZIP压缩包。`zipfile`模块为您提供了强大的功能来处理棘手的文件操作，从而为您管理和修改ZIP文件提供了无限的可能性。所以，现在就开始发挥您的Python技能，释放ZIP压缩包的全部潜能吧！