Python写入CSV文件遇KeyError:5，请求代码解析与修复

一、报错根源 #

KeyError:5说明程序在退出阶段尝试访问字典中不存在的键5。结合常见代码逻辑，大概率是退出分支里写死了对键5的直接访问，或是循环遍历字典时出现索引越界、错误固定键值的情况。

二、代码逻辑解析（基于典型末尾片段） #

假设你的程序末尾代码类似这样：

while True:
    opt = input("输入操作(A添加/D删除/Q退出): ").upper()
    if opt == 'Q':
        # 保存字典到CSV
        with open('output.csv', 'w', newline='') as f:
            writer = csv.writer(f)
            for k in my_dict:
                writer.writerow([k, my_dict[k]])
        # 错误代码：硬编码访问不存在的键5
        print("最后一条数据：", my_dict[5])
        break
    # 其他A/D操作逻辑

这段代码的问题在于：退出时强制访问键5，但如果字典里没有这个键，就会触发KeyError。保存CSV的逻辑本身是正常的，问题出在退出时的额外无效操作。

三、记事本中CSV的输出样式 #

保存后的CSV用记事本打开是纯文本格式，每行是字典的一组键值对，用逗号分隔：

• 无表头的样式：

1,张三
2,李四
3,王五

• 加表头后的样式（代码中写入writer.writerow(["序号","姓名"])）：

序号,姓名
1,张三
2,李四
3,王五

四、修复方案 #

• 删除无效的键访问代码：直接删掉退出分支里硬编码访问my_dict[5]的语句，比如例子中的print("最后一条数据：", my_dict[5])。
• 先判断键是否存在再访问：如果需要保留类似输出逻辑，先检查键是否在字典中：

  # 替换原来的硬编码访问
  if 5 in my_dict:
      print("最后一条数据：", my_dict[5])
  else:
      print("字典中不存在键5")
  

• 用get方法安全访问：不想写判断的话，用字典的get方法设置默认值，避免报错：

  print("最后一条数据：", my_dict.get(5, "无此数据"))
  

• 规范退出逻辑：确保退出时只执行必要的CSV保存操作，不要添加无关的字典访问：

  if opt == 'Q':
      # 保存CSV（可添加表头优化）
      with open('output.csv', 'w', newline='', encoding='utf-8') as f:
          writer = csv.writer(f)
          writer.writerow(["Key", "Value"])  # 添加表头
          for key, val in my_dict.items():
              writer.writerow([key, val])
      print("数据已保存，程序退出")
      break
  

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