本文完整示例代碼及文件已上傳至我的
Github
倉庫https://github.com/CNFeffery/PythonPracticalSkills
這是我的系列文章「Python實用祕技」的第13期,本系列立足於筆者日常工作中使用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))
基於這個特性,我們可以應用到很多場景下,譬如當我們希望將表格格式的字符串轉換爲pandas
數據框時,就可以像下面這樣做:
- 創建臨時目錄
前面展示了tempfile
創建臨時文件的功能,而有些場景下,我們需要創建臨時文件夾,這可以基於TemporaryDirectory()
來實現,特性類似TemporaryFile()
:
典型的應用場景是配合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]
本期分享結束,咱們下回見~👋