如何使用PyMongo将含字符串数组的CSV导入MongoDB

嘿，我懂你的困扰！你遇到的问题核心在于：CSV里的criteria列看起来是数组，但本质是带单引号的字符串伪数组，pandas读取后只会把它当成普通字符串处理，直接转成JSON或字典导入MongoDB自然还是字符串类型。咱们得先把这个字符串解析成真实的Python列表/字典对象，再导入就没问题啦。

为什么之前的方法失效？ #

你的两种方法里用了json.loads，但JSON语法要求必须用双引号，而你的criteria内容是单引号（比如{'age':35}），json.loads根本解析不了，所以这列始终是字符串。

可行解决方案 #

我们可以用Python内置的ast.literal_eval模块，它能完美处理单引号格式的Python数据结构（列表、字典都没问题）。下面是完整的步骤：

1. 导入必要的库 #

除了你原来的库，还要加上ast：

import pymongo
from pymongo import MongoClient
import pandas as pd
import ast

2. 连接MongoDB（根据你的实际配置修改） #

client = MongoClient('mongodb://localhost:27017/')  # 你的MongoDB地址
db = client['your_database']  # 替换成你的数据库名
collection = db['tmp_collection']

3. 读取CSV并解析criteria列 #

这是最关键的一步，用ast.literal_eval把字符串转成真实的Python列表：

df = pd.read_csv("file.csv")

# 对每一行的criteria列进行解析
df['criteria'] = df['criteria'].apply(lambda x: ast.literal_eval(x))

4. 转成MongoDB兼容的格式并插入 #

# 把DataFrame转成MongoDB需要的字典列表格式
records = df.to_dict('records')

# 批量插入集合
collection.insert_many(records)

可选：添加异常处理（防止格式错误的行导致崩溃） #

如果CSV里有些行的criteria格式不规范，可以加个异常处理函数：

def safe_parse_criteria(s):
    try:
        return ast.literal_eval(s)
    except (SyntaxError, ValueError):
        return []  # 格式错误时返回空列表，也可以返回None或其他默认值

df['criteria'] = df['criteria'].apply(safe_parse_criteria)

这样处理后，导入MongoDB的criteria就会是真正的数组类型，和你想要的目标格式完全一致啦！

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