如何使用 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壓縮包的全部潛能吧！