在Python中加载.pkl文件

原创于 2025-10-09 21:39:22 发布 · 338 阅读

1 ·

CC 4.0 BY-SA版权

本文为博主原创文章，转载请在显著位置标明本文出处以及作者网名，未经作者允许不得用于商业目的。

文章标签：

#python #开发语言

在Python中加载.pkl文件

.pkl文件是Python中常用的序列化文件格式，通常用于保存Python对象。你可以使用pickle模块来加载.pkl文件。以下是几种常见的方法：

基本方法

import pickle

# 方法1: 使用with语句自动处理文件关闭
with open('file.pkl', 'rb') as f:
    data = pickle.load(f)

# 方法2: 传统方式
f = open('file.pkl', 'rb')
data = pickle.load(f)
f.close()

处理大文件

对于大文件，可以使用更高效的方式：

import pickle
import gzip  # 如果文件是gzip压缩的

with gzip.open('large_file.pkl.gz', 'rb') if 'gz' in filename else open(filename, 'rb') as f:
    data = pickle.load(f)

注意事项

安全性：不要加载来自不可信来源的.pkl文件，因为pickle可以执行任意代码。
兼容性：如果文件是用Python 2保存的，而你在Python 3中加载，可能需要指定编码：

with open('file.pkl', 'rb') as f:
    data = pickle.load(f, encoding='latin1')

替代方案：对于更安全的数据交换，考虑使用joblib（特别适合大型NumPy数组）或JSON格式：

from joblib import load
data = load('file.pkl')

示例：完整流程

import pickle

# 假设我们有一个字典要保存
data_to_save = {'name': 'Alice', 'age': 30, 'scores': [85, 92, 88]}

# 保存到.pkl文件
with open('example.pkl', 'wb') as f:
    pickle.dump(data_to_save, f)

# 从.pkl文件加载
with open('example.pkl', 'rb') as f:
    loaded_data = pickle.load(f)

print(loaded_data)  # 输出: {'name': 'Alice', 'age': 30, 'scores': [85, 92, 88]}