首页 > 解决方案 > MongoDB Count 基于一个字段的输出

问题描述

假设我有以下数据集,

//listings collection

{
  "city": New York,
  "beds": 2,
  "build": concrete
}

{
  "city": New York,
  "beds": 4,
  "build": wood
}

{
  "city": New York,
  "beds": 3,
  "build": asphalt
}

{
  "city": New York,
  "beds": 1,
  "build": concrete
}

我可以通过以下查询获得以下平均床位数

        db.listings.aggregate(
            [  
                {  
                    $match: {  
                        "city": "New York"
                    }
                },
                {  
                    $group: {  
                        "_id": null,
                        "Avg-Beds": {  
                            $avg:"$beds
                         }
                    }
                }
            ])

这很酷,但我真正想要的是

{
    "Avg-Beds": 2
    "Build" {
                "Asphalt" : 1,
                "Wood": 1,
                "Concrete": 2

}

总之,我想平均床位,但我想同时计算"build"字段的输出。mongodb如何实现这一点?

更好的是类似的输出

"Build": {
           "Asphalt": "25%"
 }

这将给出一个基于百分比的值。请注意,我没有预定义的一组"build"输出字段。

标签: node.jsmongodbmongodb-querymongojs

解决方案


您可以尝试以下聚合:

db.listings.aggregate([
    {
        $match: { "city": "New York" }
    },
    {
        $group: {
            _id: null,
            avg: { $avg: "$beds" },
            docs: { $push: "$$ROOT" }
        }
    },
    {
        $unwind: "$docs"
    },
    {
        $group: {
            _id: "$docs.build",
            avg: { $first: "$avg" },
            beds: { $sum: "$docs.beds" }
        }
    },
    {
        $group: {
            _id: null,
            avg: { $first: "$avg" },
            total: { $sum: "$beds" },
            Build: { $push: { k: "$_id", v: "$beds" } }
        }
    },
    {
        $addFields: {
            Build: {
                $map: {
                    input: "$Build",
                    as: "b",
                    in: {
                        k: "$$b.k",
                        v: { $divide: [ "$$b.v", "$total" ] }
                    }
                }
            }
        }
    },
    {
        $project: {
            _id: 0,
            avg: 1,
            Build: { $arrayToObject: "$Build" }
        }
    }
])

问题是您需要多个独立的聚合,因此您可以执行第一个 ( $avg),然后将其结果嵌入到集合的每个文档中(将所有文档推送到docs字段下并展开该字段)。然后,您可以构建一个k-v对数组以应用$arrayToObject来表示百分比。

结果你会得到:

{ "avg" : 2.5, "Build" : { "asphalt" : 0.3, "wood" : 0.4, "concrete" : 0.3 } }

推荐阅读