python - Mongo 聚合 - 使用 $project 中创建的变量

Question

是否可以在同一阶段使用 $project 阶段中定义的变量？

例如，我有这个聚合管道:

pipeline = [
  { '$match': {} },
  {
    '$group': {
      '_id': '$_id',
      'n': { '$first': 'n' }
    }
  },
  {
    '$project': {
      'name': 1,
      'n': 1,
      'revenue': { '$multiply': ['$n', 2] },
      'cost': { '$multiply': ['$revenue', 0.25] }
    }
  }
]

我想在相同的 $project 阶段使用 $revenue 变量(我在 $project 中定义)来计算cost 的值，但这不起作用。

有什么方法可以轻松高效地做到这一点？我想过做几个预测，但每次我需要计算一个额外的变量时，我都需要投影很多变量(~25 个变量)，而且我有一堆变量要计算，它们相互依赖(~5 个变量)，而且我的代码可能看起来很糟糕，因为会有很多相同变量的投影。我该怎么办？

Answer 1

不，但您可以嵌套投影运算符作为变通方法:

pipeline = [{   
    '$match': {}
},
{
    '$group': {
        '_id': '$_id',
        'n': {'$first': 'n'}
},
{
    '$project': {
        'name': 1,
        'n': 1,
        'revenue': {'$multiply': ['$n', 2]},
        'cost': {'$multiply': [{'$multiply': ['$n', 2]}, 0.25]}
    }
}]

在这种情况下，您可以简化为:

pipeline = [{   
    '$match': {}
},
{
    '$group': {
        '_id': '$_id',
        'n': {'$first': 'n'}
},
{
    '$project': {
        'name': 1,
        'n': 1,
        'revenue': {'$multiply': ['$n', 2]},
        'cost': {'$multiply': ['$n', 2, 0.25]}
    }
}]