使用MongoDB聚合计算计数和平均值

时间:2012-10-20 16:27:09

标签: node.js mongodb mongoose aggregation-framework

我有一个像这样的简单数据库布局:

client
    id
    sex (male/female)
    birthday (date)    

client
    id
    sex (male/female)
    birthday (date)  

(...)

我正在尝试编写一个汇总命令,输出我有多少男性和女性客户,我也想输出男性和女性的平均年龄,不确定我是否可以这样做命令还是我需要2个独立的命令?

// Count of males/females, average age
Clients.aggregate({
    $project : {"sex"      : 1,
            "sexCount" : 1,
            "birthday" : 1,
            "avgAge"   : 1
               } 
    },
    {
        $match: {"sex": {$exists: true}} 
    },
    {
        $group: {
                    _id      : "$sex",
            sexCount : { $sum: 1 },
            avgAge   : { $avg: "$birthday" },
            }
    },
    { $sort: { _id: 1 } }
    , function(err, sex_dbres) {
            if (err)
                throw err;
            else{
                (...)
            }
        });         

根据上面的代码,我得到男性/女性的数量,但avgAge为0。任何想法?

非常感谢

2 个答案:

答案 0 :(得分:6)

如果您将年龄存储在原始文档中,答案会更简单(正如Dmitry发布的那样,您可以在avgAge:{$avg:"$age"}步骤中直接$group

聚合框架非常漂亮,并且有许多很酷的操作符,可以让你计算这个丢失的年龄字段"即时#34;。

我要将汇总的每一步存储在一个变量中,以便更容易看到正在发生的事情:

today = new Date();
// split today and bday into numerical year and numerical day-of-the-year
project1= {
    "$project" : {
        "sex" : 1,
        "todayYear" : {
            "$year" : today
        },
        "todayDay" : {
            "$dayOfYear" : today
        },
        "by" : {
            "$year" : "$bday"
        },
        "bd" : {
            "$dayOfYear" : "$bday"
        }
    }
};
// calculate age in days by subtracting bday in days from today in days
project2 = {
    "$project" : {
        "sex" : 1,
        "age" : {
            "$subtract" : [
                {
                    "$add" : [
                        {
                            "$multiply" : [
                                "$todayYear",
                                365
                            ]
                        },
                        "$todayDay"
                    ]
                },
                {
                    "$add" : [
                        {
                            "$multiply" : [
                                "$by",
                                365
                            ]
                        },
                        "$bd"
                    ]
                }
            ]
        }
    }
};
// sum up for each sex the count and compute avg age (in days)
group = {
    "$group" : {
        "_id" : "$sex",
        "total" : {
            "$sum" : 1
        },
        "avgAge" : {
            "$avg" : "$age"
        }
    }
};
// divide days by 365 to get age in years.
project3 = {
    "$project" : {
        "_id" : 0,
        "sex" : "$_id",
        "total" : 1,
        "averageAge" : {
            "$divide" : [
                "$avgAge",
                365
            ]
        }
    }
};

现在您可以运行聚合:

> db.client.find({},{_id:0})
{ "sex" : "male", "bday" : ISODate("2000-02-02T08:00:00Z") }
{ "sex" : "male", "bday" : ISODate("1987-02-02T08:00:00Z") }
{ "sex" : "female", "bday" : ISODate("1989-02-02T08:00:00Z") }
{ "sex" : "female", "bday" : ISODate("1993-11-02T08:00:00Z") }
> db.client.aggregate([ project1, project2, group, project3 ])
{
    "result" : [
        {
            "sex" : "female",
            "total" : 2,
            "averageAge" : 21.34109589041096
        },
        {
            "sex" : "male",
            "total" : 2,
            "averageAge" : 19.215068493150685
        }
    ],
    "ok" : 1
}
> 

这不简单的原因是目前Aggregation Framework不支持直接减去日期。请投票选择针对下一个主要版本的https://jira.mongodb.org/browse/SERVER-6239 - 一旦实施,它应该允许直接减去日期(尽管您仍然需要将其转换为适当的粒度,在这种情况下可能是多年)

答案 1 :(得分:4)

日期对象不能“平均”,但数字可以。您可以将日期转换为时间戳值,然后从中查找平均值。但是,这仍然不是平均年龄,您需要从聚合函数之外的当前日期中减去结果。

另一种选择是假设年龄只能使用日期的一部分来计算(也就是说,如果我出生于2000年12月1日,在今天的报告中,我将是12岁,而不是11岁)。在这种情况下,您可以使用date operators来提取年份值。

$project : {"sex"      : 1,
            "sexCount" : 1,
            "year" : {$year: "$birthday"},
           } 
},
$project : {"sex"      : 1,
            "sexCount" : 1,
            "age" : {$subtract: [2012, '$year']},
           } 
},