7

我在名为 b1.pkl 的文件中有一个腌制对象：

$ ls -l b*
-rw-r--r--  1 fireball  staff  64743950 Oct 11 15:32 b1.pkl

然后我运行以下 python 代码来加载对象并将其转储到一个新文件中：

import numpy as np
import cPickle as pkl

fin = open('b1.pkl', 'r')
fout = open('b2.pkl', 'w')

x = pkl.load(fin)
pkl.dump(x, fout)

fin.close()
fout.close()

此代码创建的文件是原来的两倍多：

$ ls -l b*
-rw-r--r--  1 fireball  staff   64743950 Oct 11 15:32 b1.pkl
-rw-r--r--  1 fireball  staff  191763914 Oct 11 15:47 b2.pkl

谁能解释一下为什么新文件比原来的大很多？它应该包含完全相同的结构。

4

3 回答 3

12

可能是原始泡菜使用了其他协议。例如，尝试将其指定protocol=2为第二个关键字参数pickle.dump并再次对其进行测试。二元泡菜的大小应该小得多。

于 2012-10-11T23:06:39.350 回答
5

您的原件很可能b1.pkl是使用更有效的协议模式（1 或 2）挑选出来的。所以你的文件开始变小了。

当您使用 cPickle 加载时，它会自动从文件中为您检测协议。但是当你用默认参数再次转储它时，它将使用更大的协议 0。它这样做是为了便携性/兼容性。您需要明确请求二进制协议。

import numpy as np
import cPickle

# random data
s = {}
for i in xrange(5000):
    s[i] = np.random.randn(5,5)

# pickle it out the first time with binary protocol
with open('data.pkl', 'wb') as f:
    cPickle.dump(s, f, 2)

# read it back in and pickle it out with default args
with open('data.pkl', 'rb') as f:
    with open('data2.pkl', 'wb') as o:
        s = cPickle.load(f)
        cPickle.dump(s, o)

$ ls -l
1174109 Oct 11 16:05 data.pkl
3243157 Oct 11 16:08 data2.pkl
于 2012-10-11T23:09:25.443 回答
3

pkl.dump(x, fout, 2) 可能会产生相同的文件大小。不指定协议版本将使pickle使用旧版本0。

于 2012-10-11T23:06:28.273 回答