「Python實用秘技13」Python中臨時檔案的妙用

費弗裡發表於2023-02-26
Python

本文完整示例程式碼及檔案已上傳至我的Github倉庫https://github.com/CNFeffery/PythonPracticalSkills

　　這是我的系列文章「Python實用秘技」的第13期，本系列立足於筆者日常工作中使用Python積累的心得體會，每一期為大家帶來一個幾分鐘內就可學會的簡單小技巧。

　　作為系列第13期，我們即將學習的是：Python中臨時檔案的妙用。

「Python實用秘技13」Python中臨時檔案的妙用

　　當我們用Python編寫程式時，有時候需要臨時儲存資料且不希望佔用多少記憶體，亦或是需要寫出檔案到檔案系統供後續程式讀取，這些情況下以建立臨時檔案的方式進行處理，既不會干擾本地檔案系統，又安全省事。

　　而透過使用Python中的標準庫tempfile，我們就可以很方便的進行臨時檔案相關操作，其主要用法有兩種：

  • 建立臨時檔案

　　tempfile的經典用法之一是使用其TemporaryFile()配合with上下文管理器，在本地檔案系統的臨時檔案目錄下建立具有隨機名稱的檔案，並且在with內的程式碼執行完成後，該臨時檔案會自動被銷燬：

import os
import tempfile

with tempfile.TemporaryFile() as f:
    print(f.name)
    print(os.path.exists(f.name))

print(os.path.exists(f.name))
「Python實用秘技13」Python中臨時檔案的妙用

　　基於這個特性，我們可以應用到很多場景下，譬如當我們希望將表格格式的字串轉換為pandas資料框時，就可以像下面這樣做：

「Python實用秘技13」Python中臨時檔案的妙用
  • 建立臨時目錄

　　前面展示了tempfile建立臨時檔案的功能，而有些場景下，我們需要建立臨時資料夾，這可以基於TemporaryDirectory()來實現，特性類似TemporaryFile()：

「Python實用秘技13」Python中臨時檔案的妙用

　　典型的應用場景是配合TemporaryDirectory()和shutil生成壓縮包檔案：

import shutil
import zipfile
import numpy as np

with tempfile.TemporaryDirectory() as p:
    
    # 模擬向當前臨時目錄下寫出多個檔案
    for i in range(100):
        (
            pd
            .DataFrame(np.random.rand(10, 3), columns=list('ABC'))
            .to_csv(os.path.join(p, f'demo{i+1}.csv'), index=False)
        )
        
    # 將當前臨時目錄轉為zip壓縮包
    shutil.make_archive('./demo', 'zip', p)

# 檢視目標壓縮檔案內的檔案
[file.filename for file in zipfile.ZipFile('demo.zip').filelist]
「Python實用秘技13」Python中臨時檔案的妙用

　　本期分享結束，我們們下回見~?

相關文章