从pickle文件,在python中恢复数据

时间:2013-07-02 21:30:52

标签: python dictionary pickle

我在使用Python中的字典恢复姿势时遇到了一些问题。我刚刚创建了一个包含对象数据的字典,并将这个字典保存在pickle文件中。

现在我想从这个pickle文件中获取所有数据并将其恢复到另一个对象。 你能告诉我什么是最好的方法吗?

这是我的save_pose函数:

def save_pose () :

     output = open('pose_dictionary.pkl', 'wb')
     pickle.dump(grab_pose(cmds.ls(selection=1), True), output)
     output.close()

     dict_file = open('pose_dictionary.pkl', 'rb')           
     dict_pkld = pickle.load(dict_file)
     dict_file.close()


     print dict_pkld

我刚刚在一个pickle文件中保存了一个字典,所以当打印dict_pkld它会返回这个

{u'left_pCube1_control': {u'translateX': {'value': 0.0, 'key': False}, u'translateY': {'value': 0.0, 'key': False}, u'translateZ': {'value': 0.0, 'key': False}, u'scaleX': {'value': 4.1730065104412066, 'key': {'outTangent': u'auto', 'inTangent': u'auto', 'inAngle': 0.0, 'outAngle': 0.0}}, u'scaleY': {'value': 4.1730065104412066, 'key': {'outTangent': u'auto', 'inTangent': u'auto', 'inAngle': 0.0, 'outAngle': 0.0}}, u'visibility': {'value': True, 'key': False}, u'rotateX': {'value': 0.0, 'key': False}, u'rotateY': {'value': 0.0, 'key': False}, u'rotateZ': {'value': 0.0, 'key': False}, u'scaleZ': {'value': 4.1730065104412066, 'key': {'outTangent': u'auto', 'inTangent': u'auto', 'inAngle': 0.0, 'outAngle': 0.0}}}

我现在想要将这些数据从这个文件恢复到一个对象。

2 个答案:

答案 0 :(得分:0)

  

我在从字典中恢复姿势时遇到了一些问题   蟒

究竟出现了什么问题......如果我不得不猜测我的grab_pose功能有问题。

当我运行以下脚本时,我没有收到任何错误。

import pickle

output = open('pose_dictionary.pkl', 'wb')
pickle.dump({"test":1}, output)
output.close()

dict_file = open('pose_dictionary.pkl', 'rb')           
dict_pkld = pickle.load(dict_file)
dict_file.close()

print dict_pkld

答案 1 :(得分:-1)

Docs on pickle.load

  

从打开的文件对象文件中读取一个字符串,并将其解释为pickle数据流,重建并返回原始对象层次结构。

在您的功能中,您既可以保存也可以还原对象。你应该拆分这些操作(并使用with):

filename = 'pose_dictionary.pkl'

def save_pose(data):
     with open(filename , 'wb') as output:
         pickle.dump(data, output)

def restore_pose():
     with open(filename , 'rb') as dict_file:
         return pickle.load(dict_file)

data = grab_pose(cmds.ls(selection=1), True) #assumption, based on your code
save_pose(data)
#... 
data = restore_pose()