numpy python 兼容_Pickle在Python 2和3之间不兼容numpy数组

我正在尝试使用此程序加载在Python 3.2中链接here的MNIST数据集：

import pickle

import gzip

import numpy

with gzip.open('mnist.pkl.gz', 'rb') as f:

l = list(pickle.load(f))

print(l)

不幸的是,它给了我错误：

Traceback (most recent call last):

File "mnist.py", line 7, in

train_set, valid_set, test_set = pickle.load(f)

UnicodeDecodeError: 'ascii' codec can't decode byte 0x90 in position 614: ordinal not in range(128)

然后我尝试在Python 2.7中解码pickle文件,并重新编码.所以,我在Python 2.7中运行了这个程序：

import pickle

import gzip

import numpy

with gzip.open('mnist.pkl.gz', 'rb') as f:

train_set, valid_set, test_set = pickle.load(f)

# Printing out the three objects reveals that they are

# all pairs containing numpy arrays.

with gzip.open('mnistx.pkl.gz', 'wb') as g:

pickle.dump(

(train_set, valid_set, test_set),

g,

protocol=2) # I also tried protocol 0.

它运行没有错误,所以我在Python 3.2中重新编写这个程序：

import pickle

import gzip

import numpy

# note the filename change

with gzip.open('mnistx.pkl.gz', 'rb') as f:

l = list(pickle.load(f))

print(l)

但是,它给了我与以前相同的错误.我如何让它工作？