加载pickle文件时出错

时间:2016-11-09 09:09:43

标签: python-3.x pickle

我在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”选项来保存/加载。任何人都知道为什么这不起作用?

1 个答案:

答案 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}}