python - PyMongo 查询返回一个嵌套的 DataFrame

Question

我对一个非常大的数据集执行聚合。这是我的代码:

pipeline = [{"$match": {"Name_md_group": "ZEITUNGEN"}}, {"$group": {"_id": {"Name_freq": "$Name_freq"}, "total": {"$sum": "$Cost"}}}]
result = pd.DataFrame(list(collection.aggregate(pipeline)))

结果:

                                          _id         total
    0  {'Name_freq': 'WOECHENTLICH FUENFMAL'}  2.074940e+07
    1               {'Name_freq': 'SONSTIGE'}  2.284889e+07
    2           {'Name_freq': 'WOECHENTLICH'}  8.522535e+07
    3               {'Name_freq': 'TAEGLICH'}  3.700943e+07
    4  {'Name_freq': 'WOECHENTLICH SECHSMAL'}  1.489394e+09

我能以某种方式取回嵌套对象吗？为什么会这样？有办法摆脱它吗？列名称 _id 应为 Name_freq。谁能帮我吗？

Answer 1

将表达式直接放入_id并添加额外的$project阶段。

pipeline = [
  { $match: { Name_md_group: "ZEITUNGEN" } },
  { $group: { _id: "$Name_freq", total: { $sum: "$Cost" } } },
  { $project: { Name_freq: "$_id", _id: 0, total: 1 }}
]