保存和加载对象以及使用pickle

Question

我正在尝试使用 pickle 模块保存和加载对象。
首先我声明我的对象：

>>> class Fruits:pass
...
>>> banana = Fruits()

>>> banana.color = 'yellow'
>>> banana.value = 30

之后我打开一个名为“Fruits.obj”的文件（之前我创建了一个新的 .txt 文件并重命名了“Fruits.obj”）：

>>> import pickle
>>> filehandler = open(b"Fruits.obj","wb")
>>> pickle.dump(banana,filehandler)

完成此操作后，我关闭了我的会话并开始了一个新会话，然后我放入了下一个会话（尝试访问它应该保存的对象）：

file = open("Fruits.obj",'r')
object_file = pickle.load(file)

但我有这样的信息：

Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "C:\Python31\lib\pickle.py", line 1365, in load
encoding=encoding, errors=errors).load()
ValueError: read() from the underlying stream did notreturn bytes

我不知道该怎么做，因为我不明白这条信息。 有谁知道我如何加载我的对象“香蕉”？ 谢谢！

编辑： 正如你们中的一些人所建议的那样，我说：

>>> import pickle
>>> file = open("Fruits.obj",'rb')

没有问题，但是我放的下一个是：

>>> object_file = pickle.load(file)

我有错误：

Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "C:\Python31\lib\pickle.py", line 1365, in load
encoding=encoding, errors=errors).load()
EOFError

Answer 1

至于你的第二个问题：

 Traceback (most recent call last):
 File "<stdin>", line 1, in <module>
 File "C:\Python31\lib\pickle.py", line
 1365, in load encoding=encoding,
 errors=errors).load() EOFError

在您读取文件内容后，文件指针将位于文件末尾 - 将不再有数据可读取。您必须倒带文件，以便再次从头开始读取：

file.seek(0)

您通常想要做的是使用上下文管理器打开文件并从中读取数据。这样，文件将在块执行完成后自动关闭，这也将帮助您将文件操作组织成有意义的块。

最后，cPickle 是 C 中 pickle 模块的更快实现。所以：

In [1]: import _pickle as cPickle

In [2]: d = {"a": 1, "b": 2}

In [4]: with open(r"someobject.pickle", "wb") as output_file:
   ...:     cPickle.dump(d, output_file)
   ...:

# pickle_file will be closed at this point, preventing your from accessing it any further

In [5]: with open(r"someobject.pickle", "rb") as input_file:
   ...:     e = cPickle.load(input_file)
   ...:

In [7]: print e
------> print(e)
{'a': 1, 'b': 2}

Answer 2

您可以使用anycache 为您完成这项工作。假设您有一个创建实例的函数myfunc：

from anycache import anycache

class Fruits:pass

@anycache(cachedir='/path/to/your/cache')    
def myfunc()
    banana = Fruits()
    banana.color = 'yellow'
    banana.value = 30
return banana

Anycache 第一次调用myfunc 并将结果腌制到一个 cachedir 中的文件，使用唯一标识符（取决于函数名和参数）作为文件名。 在任何连续运行中，都会加载腌制对象。

如果 cachedir 在 python 运行之间保留，则腌制对象取自上一次 python 运行。

函数参数也被考虑在内。 重构的实现同样有效：

from anycache import anycache

class Fruits:pass

@anycache(cachedir='/path/to/your/cache')    
def myfunc(color, value)
    fruit = Fruits()
    fruit.color = color
    fruit.value = value
return fruit

Answer 3

您似乎想跨会话保存类实例，使用pickle 是一种不错的方法。但是，有一个名为klepto 的包将对象的保存抽象为字典接口，因此您可以选择腌制对象并将其保存到文件中（如下所示），或者腌制对象并将其保存到数据库中，或使用 json 或许多其他选项代替 pickle。 klepto 的好处在于，通过抽象为一个通用接口，它变得很容易，因此您不必记住如何通过酸洗到文件或其他方式保存的低级细节。

请注意，它适用于动态添加的类属性，而 pickle 无法做到...

dude@hilbert>$ python
Python 2.7.6 (default, Nov 12 2013, 13:26:39) 
[GCC 4.2.1 Compatible Apple Clang 4.1 ((tags/Apple/clang-421.11.66))] on darwin
Type "help", "copyright", "credits" or "license" for more information.
>>> from klepto.archives import file_archive 
>>> db = file_archive('fruits.txt')
>>> class Fruits: pass
... 
>>> banana = Fruits()
>>> banana.color = 'yellow'
>>> banana.value = 30
>>> 
>>> db['banana'] = banana 
>>> db.dump()
>>> 

然后我们重新开始……

dude@hilbert>$ python
Python 2.7.6 (default, Nov 12 2013, 13:26:39) 
[GCC 4.2.1 Compatible Apple Clang 4.1 ((tags/Apple/clang-421.11.66))] on darwin
Type "help", "copyright", "credits" or "license" for more information.
>>> from klepto.archives import file_archive
>>> db = file_archive('fruits.txt')
>>> db.load()
>>> 
>>> db['banana'].color
'yellow'
>>> 

Klepto 适用于 python2 和 python3。

在此处获取代码： https://github.com/uqfoundation

Answer 4

以下对我有用：

class Fruits: pass

banana = Fruits()

banana.color = 'yellow'
banana.value = 30

import pickle

filehandler = open("Fruits.obj","wb")
pickle.dump(banana,filehandler)
filehandler.close()

file = open("Fruits.obj",'rb')
object_file = pickle.load(file)
file.close()

print(object_file.color, object_file.value, sep=', ')
# yellow, 30

Answer 5

您没有以二进制模式打开文件。

open("Fruits.obj",'rb')

应该可以。

对于您的第二个错误，该文件很可能是空的，这意味着您无意中清空了它或使用了错误的文件名等。

（这是假设您确实关闭了会话。如果没有，那是因为您没有在写入和读取之间关闭文件）。

我测试了你的代码，它可以工作。

Answer 6

您也忘记将其读取为二进制文件。

在你的写作部分你有：

open(b"Fruits.obj","wb") # Note the wb part (Write Binary)

在阅读部分你有：

file = open("Fruits.obj",'r') # Note the r part, there should be a b too

所以替换为：

file = open("Fruits.obj",'rb')

它会起作用的:)

---

至于您的第二个错误，很可能是由于未正确关闭/同步文件造成的。

试试这段代码写：

>>> import pickle
>>> filehandler = open(b"Fruits.obj","wb")
>>> pickle.dump(banana,filehandler)
>>> filehandler.close()

这个（不变的）阅读：

>>> import pickle
>>> file = open("Fruits.obj",'rb')
>>> object_file = pickle.load(file)

更简洁的版本是使用with 语句。

写作：

>>> import pickle
>>> with open('Fruits.obj', 'wb') as fp:
>>>     pickle.dump(banana, fp)

阅读：

>>> import pickle
>>> with open('Fruits.obj', 'rb') as fp:
>>>     banana = pickle.load(fp)

Answer 7

在这种情况下总是以二进制模式打开

file = open("Fruits.obj",'rb')