python - 从 pymongo 加载单个字段作为数组

Question

我有一个 mongo 对象列表，基本上我想从中提取一个字段。

假设我的对象结构如下

[
    {'name': 'Lucy', age: '24'},
    {'name': 'Nicole', age: '22'},
    {'name': 'John', age: '28'},
]

我想从 PyMongo 加载格式为 ['Lucy', 'Nicole', 'Lauren'] 的数组。

一种方法是简单的 python 列表理解:

names = [p['name'] for p in db.people.find({'age': {'$gt': 18}}, ['name'])]

这很好用，但如果有很多记录，速度会很慢。有更有效的方法吗？

Answer 1

只要您不需要包含重复项，您可以通过调用 distinct 来完成此操作在您的 find 查询返回的光标上:

names = db.people.find({'age': {'$gt': 18}}).distinct('name')