如何检查 pymongo 游标是否有查询结果

Question

我需要检查 find 语句是否返回非空查询。

我正在做的事情如下：

query = collection.find({"string": field})
if not query: #do something

然后我意识到我的if 语句从未执行过，因为find 返回一个游标，查询是否为空。

因此我检查了documentation 并找到了两种可以帮助我的方法：

1. count(with_limit_and_skip=False) 其中（来自描述）：

   返回此查询的结果集中的文档数。

   这似乎是一个很好的检查方法，但这意味着我需要计算 游标中的所有结果都知道它是否为零，对吗？有点贵？

2. retrieved 其中（来自描述）：

   到目前为止检索到的文档数。

   我在一个空查询集上对其进行了测试，它返回零，但事实并非如此 明确它的作用，我不知道它是否适合我。

那么，这是检查find() 查询是否返回空集的最佳方法（最佳实践）？上述方法之一是否适合此目的？那么性能呢？还有其他方法吗？

---

要明确一点：我需要知道查询是否为空，并且我想找到光标在性能和 Pythonic 方面的最佳方式。

Answer 1

编辑：虽然在 2014 年确实如此，但现代版本的 pymongo 和 MongoDB 已经改变了这种行为。买家当心：

.count() 是查找查询中返回的结果数的正确方法。 count() 方法不会耗尽游标的迭代器，因此您可以在迭代结果集中的项目之前安全地进行 .count() 检查。

在 MongoDB 2.4 中，count 方法的性能得到了极大的提升。唯一可能减慢您的count 的是查询是否设置了索引。要确定查询是否有索引，可以执行类似

query = collection.find({"string": field})
print query.explain()

如果您在结果中看到 BasicCursor，则您需要在您的 string 字段上为此查询建立索引。

---

编辑：正如@a​​lvapan 指出的，pymongo deprecated this pymongo 3.7+ 中的方法现在更喜欢您在单独的查询中使用count_documents。

item_count = collection.count_documents({"string": field})

计算您在查询中返回的项目数的正确方法是在遍历查询后检查查询上的.retreived 计数器，或者首先检查enumerate查询：

# Using .retrieved
query = collection.find({"string": field})
for item in query:
    print(item)

print('Located {0:,} item(s)'.format(query.retrieved))

或者，另一种方式：

# Using the built-in enumerate
query = collection.find({"string": field})
for index, item in enumerate(query):
    print(item)

print('Located {0:,} item(s)'.format(index+1))

Answer 2

使用 find_one 而不是 find 怎么样？然后你可以检查你是否得到了结果或None。而如果“string”被索引了，你可以传递fields = {"string":1, "_id" :0}，从而使它成为一个仅索引查询，这样更快。

Answer 3

根据我的测试，最快的方法是

if query.first():
    # do something

In [51]: %timeit query = MyMongoDoc.objects(); query.first()
100 loops, best of 3: 2.12 ms per loop

In [52]: %timeit query = MyMongoDoc.objects(); query.count()
100 loops, best of 3: 4.28 ms per loop

（使用 MongoDB 2.6.7，2015-03-26）

Answer 4

另一种解决方案是将光标转换为列表，如果光标没有任何数据，则为空列表，否则列表包含所有数据。

 doc_list = collection.find({}); #find all data
 have_list = True if len(list(doc_list)) else False;