假设您要保存的所有变量都是当前函数的本地变量,那么您可以通过^{}函数获取它们。这几乎总是一个非常坏的主意,但它是可行的。在
例如:def test():
a=1
b=2
pickle.dump(file, locals())
如果您print locals(),您将看到它只是一个dict,每个局部变量都有一个键。所以,当你以后load泡菜时,你会得到同样的结论。如果你想把它注入到你的本地环境中,你可以……但是你必须非常小心。例如,此函数:
^{pr2}$
…将被编译为期望a是全局的,而不是本地的,因此更新本地{}的事实将不会产生任何影响。在
这只是这样做是个坏主意的原因之一。在
那么,右翼要做什么呢?在
最简单的是,不要有一个完整的变量集合,只需要有一个包含大量键的dict。然后你就可以对这句话进行腌制和解扣,一切都是微不足道的。在
或者,也可以使用元组显式地pickle和unpickle所需的变量:def test():
a = 1
b = 2
pickle.dump(file, (a, b))
def test2():
a, b = pickle.load(file)
print a
在一个注释中,你说你想要pickle一个数组或变量,跳过任何不能被pickle的。在
为了使事情更简单,假设您实际上只想pickle一个dict,跳过任何不能pickle的值。(上述内容应说明为什么此解决方案仍然是完全通用的。)
那么,如何知道一个值是否可以被pickle?试图预测这是一个棘手的问题。即使您有一个包含所有可pickle类型的完美列表,这仍然没有帮助—一个充满整数的列表可以被pickle,但是一个充满绑定实例方法的列表却不能
{在Python中,请求原谅是一种很重要的方法。在
下面是一个简单的演示:def is_picklable(value):
try:
pickle.dumps(value)
except TypeError:
return False
else:
return True
def filter_dict_for_pickling(d):
return {key: value for key, value in d.items() if is_picklable((key, value))}
如果将整个存储过程放在包装器函数中,则可以使这一过程更简单、更高效:def pickle_filtered_dict(d, file):
for key, value in d.items():
pickle.dump((key, value), file)
except TypeError:
pass
def pickles(file):
try:
while True:
yield pickle.load(file)
except EOFError:
pass
def unpickle_filtered_dict(file):
return {key: value for key, value in pickles(file)}