使用zlib和cPickle将字典压缩/解压缩到文件
问题描述:
我正在使用python将zlib压缩和cPickled字典写入文件。它似乎工作,但是,我不知道如何重新读取文件。使用zlib和cPickle将字典压缩/解压缩到文件
我包括以下代码,其中包括我已尝试(和相关的错误消息)的几个事情。我无处可去。
import sys
import cPickle as pickle
import zlib
testDict = { 'entry1':1.0, 'entry2':2.0 }
with open('test.gz', 'wb') as fp:
fp.write(zlib.compress(pickle.dumps(testDict, pickle.HIGHEST_PROTOCOL),9))
attempt = 0
try:
attempt += 1
with open('test.gz', 'rb') as fp:
step1 = zlib.decompress(fp)
successDict = pickle.load(step1)
except Exception, e:
print "Failed attempt:", attempt, e
try:
attempt += 1
with open('test.gz', 'rb').read() as fp:
step1 = zlib.decompress(fp)
successDict = pickle.load(step1)
except Exception, e:
print "Failed attempt:", attempt, e
try:
attempt += 1
with open('test.gz', 'rb') as fp:
step1 = zlib.decompress(fp.read())
successDict = pickle.load(step1)
except Exception, e:
print "Failed attempt:", attempt, e
try:
attempt += 1
with open('test.gz', 'rb') as fp:
d = zlib.decompressobj()
step1 = fp.read()
step2 = d.decompress(step1)
step3 = pickle.load(step2)
except Exception ,e:
print "Failed attempt:", attempt, e
try:
attempt += 1
with open('test.gz', 'rb') as fp:
d = zlib.decompressobj()
step1 = fp.read()
step2 = d.decompress(step1)
step3 = pickle.load(step2)
except Exception ,e:
print "Failed attempt:", attempt, e
我收到以下错误:
Failed attempt: 1 must be string or read-only buffer, not file
Failed attempt: 2 __exit__
Failed attempt: 3 argument must have 'read' and 'readline' attributes
Failed attempt: 4 argument must have 'read' and 'readline' attributes
Failed attempt: 5 argument must have 'read' and 'readline' attributes
但愿这只是一些显而易见的(给别人)解决这个问题,我只是缺少。谢谢你的帮助!
答
你正在尝试次数3-5的错误是因为您使用pickle.load
代替pickle.loads
。前者需要一个类似文件的对象,而不是从解压缩调用中获得的字节字符串。
这将工作:
with open('test.gz', 'rb') as fp:
data = zlib.decompress(fp.read())
successDict = pickle.loads(data)