无法使用 pickle.load() 方法读取附加数据

Question

我编写了两个脚本 Write.py 和 Read.py。

Write.py 以附加模式打开 friends.txt 并获取 name、email 、phone 的输入no 然后使用 pickle.dump() 方法将字典转储到文件中，并且在此脚本中一切正常。

Read.py 以读取模式打开 friends.txt，然后使用 pickle.load() 方法将内容加载到字典中，并显示字典。

主要问题是在 Read.py 脚本中，它只显示旧数据，它从不显示附加数据？

Write.py

#!/usr/bin/python

import pickle

ans = "y"
friends={}
file = open("friends.txt", "a")
while ans == "y":
    name = raw_input("Enter name : ")
    email = raw_input("Enter email : ")
    phone = raw_input("Enter Phone no : ")

    friends[name] = {"Name": name, "Email": email, "Phone": phone}

    ans = raw_input("Do you want to add another record (y/n) ? :")

pickle.dump(friends, file)
file.close()

Read.py

#!/usr/bin/py

import pickle

file = open("friends.txt", "r")

friend = pickle.load(file)

file.close()

for person in friend:
    print friend[person]["Name"], "\t", friend[person]["Email"] , "\t", friend[person]["Phone"]

一定是什么问题，代码看起来没问题。有人能给我指出正确的方向吗？

谢谢。

Answer 1

您必须多次从文件中加载。每个写入过程都会忽略其他过程，因此它会创建一个独立于文件中其他过程的可靠数据块。如果你事后读它，它一次只读一个块。所以你可以尝试：

import pickle

friend = {}
with open('friends.txt') as f:
    while 1:
        try:
            friend.update(pickle.load(f))
        except EOFError:
            break # no more data in the file

for person in friend.values():
    print '{Name}\t{Email}\t{Phone}'.format(**person)

Answer 2

每次调用 pickle.dump 时，都必须调用 pickle.load 一次。您编写的例程不会向字典添加条目，而是添加另一个字典。您必须调用 pickle.load 直到读取整个文件，但这将为您提供多个必须合并的字典。更简单的方法是将值存储为 CSV 格式。这就像

with open("friends.txt", "a") as file:
    file.write("{0},{1},{2}\n".format(name, email, phone))

将值加载到字典中一样简单：

with open("friends.txt", "a") as file:
    friends = dict((name, (name, email, phone)) for line in file for name, email, phone in line.split(","))