向MongoDB写入含numpy布尔对象的文档时报BSON编码错误如何解决

PyMongo 默认的BSON编码器仅支持Python原生数据类型，无法识别numpy.bool_等numpy自定义类型，才会抛出你遇到的bson.errors.InvalidDocument报错。

除手动逐个转换类型外，还有以下两类更高效的方案：

• 方案1：注册自定义BSON编码器（全局生效，推荐长期使用）
  你可以给PyMongo注册numpy类型的转换规则，配置完成后所有写入操作都会自动完成类型转换，无需额外处理单条数据，示例代码如下：

  import numpy as np
  from bson.codec_options import TypeEncoder, CodecOptions
  
  # 定义numpy布尔类型编码器
  class NumpyBoolEncoder(TypeEncoder):
      python_type = np.bool_
      def transform_python(self, value):
          return bool(value)
  
  # 可按需扩展其他numpy类型的编码器，比如numpy数值类型
  class NumpyIntEncoder(TypeEncoder):
      python_type = np.integer
      def transform_python(self, value):
          return int(value)
  
  class NumpyFloatEncoder(TypeEncoder):
      python_type = np.floating
      def transform_python(self, value):
          return float(value)
  
  # 把编码器注册到PyMongo配置中
  codec_options = CodecOptions(type_registry=[
      NumpyBoolEncoder(),
      NumpyIntEncoder(),
      NumpyFloatEncoder()
  ])
  # 初始化集合实例时传入配置即可
  collection = db.get_collection("你的集合名称", codec_options=codec_options)
  

• 方案2：写入前递归转换字典内所有numpy类型（灵活适配临时场景）
  如果不想修改全局PyMongo配置，可以写一个通用递归函数，写入前对整份文档做一次类型转换即可：

  import numpy as np
  
  def convert_numpy_obj(obj):
      # 处理numpy布尔类型
      if isinstance(obj, np.bool_):
          return bool(obj)
      # 处理numpy整数类型
      elif isinstance(obj, np.integer):
          return int(obj)
      # 处理numpy浮点数类型
      elif isinstance(obj, np.floating):
          return float(obj)
      # 处理numpy数组类型
      elif isinstance(obj, np.ndarray):
          return obj.tolist()
      # 递归处理字典
      elif isinstance(obj, dict):
          return {k: convert_numpy_obj(v) for k, v in obj.items()}
      # 递归处理列表
      elif isinstance(obj, list):
          return [convert_numpy_obj(item) for item in obj]
      # 其他类型直接返回
      else:
          return obj
  
  # 写入前调用转换函数
  processed_doc = convert_numpy_obj(待写入的原字典)
  collection.insert_one(processed_doc)
  

小提示：如果你的文档是从Pandas DataFrame转换而来，可以先调用df = df.convert_dtypes()将numpy类型批量转换为Python原生类型后再生成字典写入。

内容的提问来源于stack exchange，提问作者DaneOH-89