Python中出现KeyError: 'labels'报错求助

问题场景 #

执行代码：

labels = data['labels'].reshape(-1).astype(np.int32)

时触发报错：

Exception has occurred: KeyError
'labels'
KeyError: 'labels'

原因分析 #

该错误明确表示你访问的类字典结构（字典、pandas DataFrame、numpy结构化数组等）data中，不存在'labels'这个键/列名。

排查与解决步骤 #

• 确认data的结构与有效键名
  根据data的类型打印所有可用键/列，验证是否存在'labels'：

  # 字典类型
  print(data.keys())
  # pandas DataFrame类型
  print(data.columns)
  # numpy结构化数组类型
  print(data.dtype.names)
  

  检查是否存在拼写错误（比如大小写差异、多/少字符），或者字段名本身就不是'labels'。

• 核对数据源
  如果data来自文件（CSV、JSON等），直接打开文件确认是否包含labels字段；如果是接口返回数据，查看接口文档确认字段命名是否正确。

• 处理字段缺失的情况
  若确认数据源确实没有labels字段，根据业务逻辑调整：

  • 补充对应字段的数据；
  • 将代码中的'labels'替换为正确的字段名；
  • 允许字段缺失时，用get方法避免报错，示例：

    labels = data.get('labels', np.array([])).reshape(-1).astype(np.int32)
    

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