在Python中使用pickle.dump

Question

我正在尝试学习如何在Python中使用pickle模块：

import pickle
x = 123
f = open('data.txt','w')
pickle.dump(x,f)

这是我得到的：

Traceback (most recent call last):
  File "D:\python\test.py", line 5, in <module>
    pickle.dump(x,f)
TypeError: must be str, not bytes

但是，这段代码很好用：

import pickle
dump = pickle.dump(123)
print(dump)

我做错了什么？

Answer 1

问题是您是在文本模式下打开文件。你需要在这里使用二进制文件：

>>> f = open('data.txt','w')
>>> pickle.dump(123,f)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: must be str, not bytes
>>> 
>>> f = open('data.txt','wb')
>>> pickle.dump(123,f)
>>> 

Answer 2

类文件对象的write方法，只接受单个字符串参数。 pickle模块中的dumps方法自动将参数转换为字符串，而dump方法将对象的pickle表示写入打开的文件。由于123不是字符串，因此会抛出TypeError错误。

这在 pickle.dump documentation中得到承认。