我在python3中加载pickle文件时遇到问题。请参阅以下代码:
#!/usr/bin/env python3
import csv,operator
import pickle
import os.path
pathToBin = "foo/bar/foobar.bin"
pathToCSV = "foo/bar/foobar.csv"
if os.path.isfile(pathToBin):
print("Binary file already on harddrive, loading from there")
transactions = pickle.loads( open( pathToBin, "rb" ))
else:
csvIn = open(pathToCSV,'r')
reader = csv.reader(csvIn)
header = next(reader)
header = header[0].split(";")
print("Reading file")
transactions = []
for row in reader:
# read file.
# transactions contains now lists of strings: transactions = [ ["a","b","c"], ["a2","b2","c3"], ...]
print("Dumping python file to harddrive")
myPickleFile = open(pathToBin,'wb')
pickle.dump(transactions, myPickleFile, protocol=pickle.HIGHEST_PROTOCOL)
# do some more computation
保存文件没有任何问题。但是,加载它会给我以下错误:
transactions = pickle.loads( open( pathToBin, "rb" ))
TypeError: '_io.BufferedReader' does not support the buffer interface
由于我使用python3,因此字符串的处理方式不同。因此,我特意给出了“b”选项来保存/加载。任何人都知道为什么这不起作用?
答案 0 :(得分:4)
您想使用transactions = pickle.load(open( pathToBin, "rb" ))
:
loads
从您打开的文件句柄中读取。 bytes
接受transactions = pickle.loads(open( pathToBin, "rb" ).read())
而不是文件句柄(意思是:加载&#34;字符串&#34;,现在加载&#34;字节&#34;在python 3升级后对字符串/字节处理):< / p>
with
从文件返回的字节读取。
在这种情况下,我会为你推荐第一个选项。第二种选择适用于更复杂的情况。
除此之外:最好使用with open( pathToBin,"rb") as f:
transactions = pickle.load(f)
上下文来控制文件何时关闭
{{1}}