mongodb - 字段上的 mongo 聚合总和，而字段的值应根据另一个字段映射到 +/-

Question

我有以下collection在mongodb ,

doc_id    amount    memo
1         1.0       P
1         2.0       N
2         3.0       P
2         4.0       P
2         5.0       N

我想要sum amount字段 P在memo表示正值，N表示 amount 的值为负值;所以当我 group doc_id并执行 sum ，我会得到-1.0对于 doc_id 1 ，和2.0对于 doc_id 2 。我知道如何按 doc_id 分组然后sum ,

db.collection.aggregate({ $group: { _id : 'doc_id', sum: { $sum: "$amount"}}});

但不知道如何取P和N在这里考虑一下。

Answer 1

使用$cond根据备忘录将值投影为负/正，然后将金额相加。 以下用 pymongo 编写的查询将适合您。

db.collection..aggregate([{'$project': 
                           {'amount': {'$cond': { 'if': { '$eq':['$memo','P'] }, 'then': '$amount', 
                                                'else': {'$multiply':['$amount',-1]} }},
                            'doc_id': '$doc_id'}},
                         {'$group': {'_id': '$doc_id','sum':{'$sum': '$amount'}}}])