python - mongo的分组结果到pandas DataFrame

Question

是否有任何方便的方法可以将 mongodb 组操作的结果加载到 pandas DataFrame 中？

这是 mongo 查询的分组类型:

    '$group': {
        '_id': {'country': '$country', 'oem':'$oem'},
         'count': {'$sum': 1}
    }

这将作为字典列表返回，如下所示:

[
   ...
   { "_id" : { "country" : "US", "oem" : "mmm" }, "count" : 595 },
   ...
]

我希望将其加载到 DataFrame 中，以便 country 和 oem 自动成为索引。除了重新映射结果之外，Pandas API 中是否有任何内容可以处理此问题？或者我可以以某种方式重写 mongo 查询，以便它返回一个对 pandas API 更友好的结构吗？

Answer 1

您可以使用json_normalize() :

In [59]: l
Out[59]:
[{'_id': {'country': 'UA', 'oem': 'uuuu'}, 'count': 555},
 {'_id': {'country': 'US', 'oem': 'aaaa'}, 'count': 595},
 {'_id': {'country': 'DE', 'oem': 'bbbb'}, 'count': 777}]

In [60]: from pandas.io.json import json_normalize

In [61]: l
Out[61]:
[{'_id': {'country': 'UA', 'oem': 'uuuu'}, 'count': 555},
 {'_id': {'country': 'US', 'oem': 'aaaa'}, 'count': 595},
 {'_id': {'country': 'DE', 'oem': 'bbbb'}, 'count': 777}]

In [62]: json_normalize(l)
Out[62]:
  _id.country _id.oem  count
0          UA    uuuu    555
1          US    aaaa    595
2          DE    bbbb    777

设置:

l = [
   { "_id" : { "country" : "UA", "oem" : "uuuu" }, "count" : 555 },
   { "_id" : { "country" : "US", "oem" : "aaaa" }, "count" : 595 },
   { "_id" : { "country" : "DE", "oem" : "bbbb" }, "count" : 777 },
]