先分组，使 bucketauto 在 mongodb 聚合中排在第二位

Question

我有一个结构如下的数据集：

 {
          "id": 1230239,
          "group_name": "A",
          "confidence": 0.14333882876354542,

  },
  {
          "id": 1230240,
          "group_name": "B",
          "confidence": 0.4434535,

   },

等等。

使用$bucketauto来计算每个置信度级别中的存储桶和项目数非常简单：

{
  "$bucketAuto": {
     "groupBy": "$confidence",
     "buckets": 4
 }
}

但是我怎样才能分别为每个组做同样的事情呢？

我试过这个：

    {"$group": {
                    "_id": "group",
                    "data": {
                        "$push": {
                            "confidence": "$confidence",
                        }
                    }
                }
                },
                {
                    "$bucketAuto": {
                        "groupBy": "$data.confidence",
                        "buckets": 4
                    }
                }

但这不起作用。

我大致需要的是一个 output：

{ 'groupA': 
     {
            "_id": {
                "min": 0.0005225352581638143,
                "max": 0.2905137273072962
            },
            "count": 67
        },
        {"_id": {
                "min": 0.2905137273072962,
                "max":0.5531611756507283,
            },
            "count": 43
        },
}, 
{ 'groupB': 
     {
       "_id": {
                "min": 0.0005225352581638143,
                "max": 0.2905137273072962
            },
            "count": 67
        },
        {"_id": {
                "min": 0.2905137273072962,
                "max":0.5531611756507283,
            },
            "count": 43
        },
}

任何建议或提示将不胜感激

Answer 1

$facet救援——“多组”运算符。 这个管道：

db.foo.aggregate([
{$facet: {
  "groupA": [
{$match: {"group_name": "A"}}
,{$bucketAuto: {
        "groupBy": "$confidence",
        "buckets": 3
    }}
           ]

  ,"groupB": [
{$match: {"group_name": "B"}}
,{$bucketAuto: {
        "groupBy": "$confidence",
        "buckets": 3
    }}
           ]

    }}
   ]);

产生您寻求的 output ：

{
    "groupA" : [
        {
            "_id" : {
                "min" : 0.14333882876354542,
                "max" : 0.34333882876354543
            },
            "count" : 2
        },
        {
            "_id" : {
                "min" : 0.34333882876354543,
                "max" : 0.5433388287635454
            },
            "count" : 2
        },
        {
            "_id" : {
                "min" : 0.5433388287635454,
                "max" : 0.5433388287635454
            },
            "count" : 1
        }
    ],
    "groupB" : [
        {
            "_id" : {
                "min" : 0.5433388287635454,
                "max" : 0.7433388287635454
    // etc. etc. 

如果您想要 go 完全动态，您需要分两遍进行：首先获取不同的组名称，然后从这些名称构建$facet表达式：

db.foo.distinct("group_name").forEach(function(name) {
        fct_stage["group" + name] = [
                                     {$match: {"group_name": name}}
                                     ,{$bucketAuto: {
                                             "groupBy": "$confidence",
                                             "buckets": 3
                                         }}
                                     ];
    });

db.foo.aggregate([ {$facet: fct_stage} ]);