Python中临时文件的妙用

最新推荐文章于 2025-10-12 10:35:52 发布

原创最新推荐文章于 2025-10-12 10:35:52 发布 · 628 阅读

19 ·

CC 4.0 BY-SA版权

文章标签：

#python #开发语言 #数据分析

python脚本同时被 2 个专栏收录

21 篇文章

订阅专栏

Pytho教程

16 篇文章

订阅专栏

本文介绍了Python中tempfile模块的临时文件和临时目录创建功能，以及它们在数据转换和文件压缩中的应用。同时提供了Python学习资源链接，包括安装包、Web开发、爬虫、数据分析等内容。

我们今天即将学习的是：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]