【468】存储数据变量 — pickle 模块

参考:python pickle 模块的使用

  用于 python 中特有的类型数据存储。接触这个模块是看深度学习的时候,将训练后的权重参数通过 pickle 存储,然后之后再用的时候可以通过 pickle 在提取,很方便。主要是四个方法:

  • pickle.dump(obj, file, protocol=None,)
    • 必填参数obj表示将要封装的对象
    •  必填参数file表示obj要写入的文件对象,file必须以二进制可写模式打开,即“wb”
    •  可选参数protocol表示告知pickler使用的协议,支持的协议有0,1,2,3,默认的协议是添加在Python 3中的协议3。
  • pickle.load(file,*,fix_imports=True, encoding="ASCII", errors="strict")
    • 必填参数file必须以二进制可读模式打开,即“rb”
    • 其他都为可选参数
  • pickle.dumps(obj):以字节对象形式返回封装的对象,不需要写入文件中
  • pickle.loads(bytes_object):从字节对象中读取被封装的对象,并返回

☀☀☀<< 举例 >>☀☀☀

dumps 功能

# dumps功能
import pickle
data = ['aa', 'bb', 'cc']
# dumps 将数据通过特殊的形式转换为只有python语言认识的字符串
p_str = pickle.dumps(data)
print(p_str)            7 b'x80x03]qx00(Xx02x00x00x00aaqx01Xx02x00x00x00bbqx02Xx02x00x00x00ccqx03e.

loads功能

# loads功能
# loads  将pickle数据转换为python的数据结构
mes = pickle.loads(p_str)
print(mes)
['aa', 'bb', 'cc']

dumps 功能

# dump功能
# dump 将数据通过特殊的形式转换为只有python语言认识的字符串,并写入文件
with open('D:/tmp.pk', 'wb') as f:
    pickle.dump(data, f)

loads 功能  

# load功能
# load 从数据文件中读取数据,并转换为python的数据结构
with open('D:/tmp.pk', 'rb') as f:
    data = pickle.load(f)

也可以不用 with 结构

>>> m = {'name':'alex', 'age':18}      
>>> pickle.dump(m, open('D:/deep learning/tmp/m.pkl', 'wb'))
>>> mm = pickle.load(open('D:/deep learning/tmp/m.pkl', 'rb'))   
>>> mm     
{'name': 'alex', 'age': 18}

  

原文地址:https://www.cnblogs.com/alex-bn-lee/p/12987264.html