在 python 中使用 pickle 和 for 循环从文件中读取

Question

我有一个文件，其中转储了大量列表。现在我想将此文件加载到内存中并使用其中的数据。我尝试使用“pickle”的“load”方法加载我的文件，但是，由于某种原因，它只给了我文件中的第一项。实际上我注意到它只将我的第一个列表加载到内存中，如果我想加载整个文件（多个列表），那么我必须迭代我的文件并在每个文件中使用“pickle.load（文件名）”我采取的迭代。 问题是我不知道如何用循环（for 或 while）实际实现它，因为我不知道何时到达文件末尾。 一个例子会对我有很大帮助。 谢谢

Answer 1

这个怎么样：

lists = []
infile = open('yourfilename.pickle', 'r')
while 1:
    try:
        lists.append(pickle.load(infile))
    except (EOFError, UnpicklingError):
        break
infile.close()