Python使用pickle加载字典仅打印首个键值对的问题求助

你的核心问题在于保存字典的方式错误：

• 选项2中，你每次向文件追加dump一个仅含新条目的小字典，导致文件里存储了多个独立的pickle对象，而非一个完整的大字典。
• 选项1使用pickle.load()时，只会读取文件中的第一个pickle对象（也就是你第一次保存的那个单条目字典），所以只能看到第一个条目。

解决方案 #

正确的做法是每次操作都先加载完整的字典（文件存在时），修改后再将整个字典重新dump回文件，而不是追加单个条目。同时处理文件不存在的异常情况。

修改后的完整代码 #

import pickle
from pathlib import Path

print("""
 
      1- View codes
      2- Add code
      3- Create file/add first""")

# 定义文件路径，方便复用
FILE_PATH = "dcoutput"

while True:
    choice = input("What is your choice? ")
    
    if choice == "1":
        # 检查文件是否存在
        if not Path(FILE_PATH).exists():
            print("No entries found. Please create a file first.")
            continue
        # 加载完整字典
        with open(FILE_PATH, 'rb') as infile:
            dc_message = pickle.load(infile)
        # 打印所有条目
        for key, value in dc_message.items():
            print(f"{key} : {value}")
            
    elif choice == "2":
        dc_code = input("What is the new DCcode? ")
        message = input("What is the new message? ")
        # 先加载现有字典（如果存在）
        if Path(FILE_PATH).exists():
            with open(FILE_PATH, 'rb') as infile:
                dc_message = pickle.load(infile)
        else:
            dc_message = {}
        # 添加新条目
        dc_message[dc_code] = message
        # 重新保存完整字典
        with open(FILE_PATH, 'wb') as outfile:
            pickle.dump(dc_message, outfile)
        # 打印新增内容
        print(f"Added: {dc_code} : {message}")
    
    elif choice == "3":
        # 选项3和选项2逻辑合并，创建文件本质也是添加第一个条目
        dc_code = input("What is the new DCcode? ")
        message = input("What is the new message? ")
        dc_message = {dc_code: message}
        with open(FILE_PATH, 'wb') as outfile:
            pickle.dump(dc_message, outfile)
        print(f"Created file and added: {dc_code} : {message}")
    
    else:
        print("Invalid choice. Please enter 1, 2, or 3.")

关键修改点说明 #

• 使用with语句管理文件，自动处理关闭操作，更安全简洁。
• 每次添加条目时，先加载现有完整字典，修改后再整体保存，确保文件中始终只有一个完整的字典对象。
• 选项3与选项2逻辑合并，避免重复代码（即使用户选3时文件已存在，也会正确添加条目）。
• 增加文件存在性检查，避免加载不存在的文件时抛出异常。

内容的提问来源于stack exchange，提问作者gilley178