时间戳

首页 > 解决方案 > PyMongo 查询给了我一个嵌套的 DataFrame

python - PyMongo 查询给了我一个嵌套的 DataFrame

问题描述

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

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. 谁能帮我？

标签： pythonpython-3.xmongodbaggregation-frameworkpymongo

解决方案

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

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

推荐阅读