MongoEngine - 聚合

术语"聚合"用于处理数据并返回计算结果的操作。 对集合中文档的一个或多个字段进行求和、计数和平均可以称为聚合函数。

MongoEngine 提供了aggregate() 函数封装了PyMongo 的聚合框架。 聚合操作使用集合作为输入并返回一个或多个文档作为结果。

MongoDB 使用数据处理管道的概念。 管道可以有多个阶段。 基本阶段提供过滤器并像查询一样操作。 其他工具提供按一个或多个字段进行分组和/或排序的工具、字符串连接任务、数组聚合工具等。

在 MongoDB 管道创建中定义了以下阶段 −

聚合表达式使用字段路径访问输入文档中的字段。 要指定字段路径，请使用以美元符号 $$$ 作为字段名称前缀的字符串。 表达式可以使用一个或多个布尔运算符（$and、$or、$not）和比较运算符（$eq、$gt、$lt、$gte、$lte 和 $ne）。

以下算术表达式也用于聚合 −

下面的字符串表达式也可以用于聚合 −

为了演示 aggregate() 函数在 MongoEngine 中的工作原理，让我们首先定义一个名为 orders 的文档类。

from mongoengine import *
con=connect('mydata')

class orders(Document):
   custID = StringField()
   amount= IntField()
   status = StringField()

然后我们在 orders 集合中添加以下文档 −

仅当状态等于"A"时，aggregate() 函数才用于查找每个 custID 的 amount 字段的总和。 因此，管道构造如下。

管道中的第一阶段使用 $match 来过滤状态为"A"的文档。 第二阶段使用 $group 标识符对 CustID 上的文档进行分组并执行 amount 求和。

 pipeline = [
{"$match" : {"status" : "A"}},
{"$group": {"_id": "$custID", "total": {"$sum": "$amount"}}}
]

此管道现在用作 aggregate() 函数的参数。

docs = orders.objects().aggregate(pipeline)

我们可以使用 for 循环遍历文档游标。 完整代码如下 −

from mongoengine import *
con=connect('mydata')

class orders(Document):
   custID = StringField()
   amount= IntField()
   status = StringField()

pipeline = [
   {"$match" : {"status" : "A"}},
   {"$group": {"_id": "$custID", "total": {"$sum": "$amount"}}}
   ]
docs = orders.objects().aggregate(pipeline)
for doc in docs:
   print (x)

对于给定的数据，生成以下输出 −

{'_id': 'B212', 'total': 400}
{'_id': 'A123', 'total': 750}