Python pickle模块用法实例

咖啡半糖

发布时间：2026-10-12| 2人浏览过

原创

Python 的 pickle 模块提供了灵活的数据序列化和反序列化方案，适合将程序运行时生成的对象信息永久保存到本地文件。通过掌握 pickle.dump() 与 pickle.load() 的核心接口及参数设置，配合正确的二进制文件打开模式，即可实现高效的数据读写。本文结合标准代码演示完整操作流程，并补充关键的安全与兼容性提醒。

掌握 pickle 的核心数据接口

pickle 模块主要依赖两个基础方法完成内存到磁盘的数据流转。序列化对应 pickle.dump(obj, file, [,protocol])，用于将 Python 对象 obj 按照指定的协议版本写入类文件对象 file；反序列化对应 pickle.load(file)，负责从已打开的文件流中读取字节并还原为原始 Python 对象。

在配置 dump 时，protocol 参数决定编码特征。可选值包含：0 代表 ASCII 协议（可读性强但体积偏大），1 为传统二进制协议，2 为 Python 2.3 引入的高效二进制协议（性能最优）。当 protocol 大于等于 1 时，目标文件句柄必须以二进制模式初始化，否则可能触发类型错误或数据截断。

第1步：配置写入环境并执行 dump()

目的：创建持久化存储介质，将字典与包含自引用的列表对象序列化至磁盘。

入口：在脚本作用域内直接定位目标工作目录。

动作：调用内置 open() 建立写入通道，将数据对象逐次传入 dump() 进行落盘。

重要参数：文件标识符固定使用 'wb' 开启二进制写入；protocol 依需求设定为 0（兼容老版本）或 -1（启用最高级协议）。

完成状态：控制台静默返回，当前目录下生成同名 .pkl 文件，文件大小随数据量稳步增长。

# 定义待保存的测试数据结构
import pickle

data1 = {'a': [1, 2.0, 3, 4+6j], 'b': ('string', u'Unicode string'), 'c': None}
selfref_list = [1, 2, 3]
selfref_list.append(selfref_list)

# 使用 pickle 将数据对象保存到文件
output = open('data.pkl', 'wb')
pickle.dump(data1, output, protocol=0)
pickle.dump(selfref_list, output, protocol=-1)
output.close()

第2步：读取文件流并重构对象

目的：恢复上一轮程序保存的状态，验证序列化数据的完整性与层级结构。

入口：同一路径下重新挂载文件句柄。

动作：切换至读取模式后，按写入先后顺序连续调用 load() 提取对象，最后安全释放句柄。

重要参数：文件标识符必须替换为 'rb' 匹配二进制读取；无需额外协议参数，load 会自动识别头部元数据。

完成状态：终端逐行打印校验后的键值对与嵌套列表，结构与输入完全一致且未发生类型转换丢失。

# 使用 pickle 从文件中重构 python 对象
import pprint, pickle

pkl_file = open('data.pkl', 'rb')

data1 = pickle.load(pkl_file)
pprint.pprint(data1)

data2 = pickle.load(pkl_file)
pprint.pprint(data2)

pkl_file.close()

运行核对与排错建议

模式不匹配：写入阶段误用文本模式（'w'）配合高级协议，将导致 OSError 或乱码。务必确保 dump 与 load 阶段的字符编码标志严格对齐（均使用 wb/rb）。
读取顺序错位：多次 load() 必须严格对应 dump() 的调用次序。跳读中间节点会抛出 EOFError 或引发后续对象的反解失败。
特殊对象拦截：pickle 无法处理文件指针、网络套接字、线程锁或未实现特定方法的自定义类实例，强行写入将直接中断流程并返回 TypeError。

安全限制与最佳实践

pickle.load() 底层会在反序列化期间动态执行字节码指令。因此绝对禁止对未知来源的 pkl 文件执行 load 操作，以免引发远程代码执行漏洞。日常交换纯文本结构建议优先采用 JSON 格式；仅在需要深度克隆复杂应用状态或保存神经网络权重时，才将 pickle 作为底层持久化手段。