Python實用技巧之臨時文件如何使用

本篇內容主要講解“Python實用技巧之臨時文件如何使用”，感興趣的朋友不妨來看看。本文介紹的方法操作簡單快捷，實用性強。下面就讓小編來帶大家學習“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數據框時，就可以像下面這樣做：

raw_string='''a,b,c
1,1,1
2,2,2
3,3,3
4,4,4
'''

with tempfile.TemporaryFile('w+') as f:
    f.write(raw_string)
    f.seek(0) #將指針移回文件開頭
    
    demo_df=pd.read_CSV(f)
    
demo_df

創建臨時目錄

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

with tempfile.TemporaryDirectory() as p:
    
    print(p)
    print(os.path.exists(p))
    
print(os.path.exists(p))

典型的應用場景是配合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實用技巧之臨時文件如何使用”有了更深的了解，不妨來實際操作一番吧！這里是億速云網站，更多相關內容可以進入相關頻道進行查詢，關注我們，繼續學習！