如何計算pymongo聚合遊標而不重複

Question

我想獲得pymongo 3.0+版中聚合遊標中的記錄總數。有什麼辦法可以在沒有迭代遊標的情況下獲得總數？

cursor = db.collection.aggregate([{"$match": options},{"$group": {"_id": groupby,"count": {"$sum":1}}} ]) 
cursorlist = [c for c in cursor] 
print len(cursorlist) 

有什麼辦法可以跳過上面的迭代嗎？

Answer 1

您可以添加另一個組管道，您可以在其中指定一個_id值None來計算所有輸入文檔的累計值，這是您可以獲得總計數以及原始分組計數的位置，儘管在累積陣列中：

>>> pipeline = [ 
...  {"$match": options}, 
...  {"$group": {"_id": groupby, "count": {"$sum":1}}}, 
...  {"$group": {"_id": None, "total": {"$sum": 1}, "details":{"$push":{"groupby": "$_id", "count": "$count"}}}} 
... ] 
>>> list(db.collection.aggregate(pipeline))