alex_bn_lee

导航

【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'\x80\x03]q\x00(X\x02\x00\x00\x00aaq\x01X\x02\x00\x00\x00bbq\x02X\x02\x00\x00\x00ccq\x03e.

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}

  

posted on 2020-05-29 14:24  McDelfino  阅读(317)  评论(0编辑  收藏  举报