pickle使用

一、使用pickle

pickle可以用来保存numpy数据类型。

1.1 保存

import pickle
import numpy as np 

array_a = np.array([[0.1, 0.2, 0.11], 
					[0.11, 0.22, 0.32], 
					[0.12, 0.1, 0.1]])

with open("./embed", 'wb') as fw: 
	# 注意: 使用pickle必须包含 'b'
	pickle.dump(array_a, fw)

注意：必须是以’b’二进制读写文件

1.2 加载

import pickle

with open("./embed", 'rb') as fr:
	array_a = pickle.load(fr)

二、遇到问题

2.1 EOFError: Ran out of input.

原因是读取的文件为空