在python中使用pickle和for循環從文件中讀取

Question

我有一個文件，我已經轉儲了大量的lists.Now我想將此文件加載到內存並使用其中的數據。我試圖使用“pickle”的“加載”方法加載我的文件，但是，出於某種原因，它只是給了我文件中的第一項。 實際上我注意到它只將我的第一個列表加載到內存中，如果我想加載我的整個文件（多個列表），那么我必須迭代我的文件並在每個文件中使用“pickle.load（filename）”迭代我采取。 問題是我不知道如何用循環（for或while）實際實現它，因為我不知道何時到達文件的末尾。 一個例子對我幫助很大。 謝謝

Answer 1

這個怎么樣：

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