从pickle文件加载前n个项目

时间:2017-08-11 04:01:36

标签: python

这可能是一件不寻常的事情,但我想从pckle文件中加载n个项目,它可以是随机的,也可以是第一个或最后一个项目。

原因是我正在测试一个神经网络,它将pickle文件作为数据加载,我希望网络在测试阶段运行得更快(从100000项减少到n)。

现在我正在加载泡菜:

l = pickle.load(open(file, 'rb'))

是否有任何参数只能加载前n个项目?

该文件如下所示:

{{3}}

1 个答案:

答案 0 :(得分:-1)

实际上这很容易:

n = 100;
l = pickle.load(open(file, 'rb'))[1:n];