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

Question

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

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。有什么想法吗？

非常感谢

Answer 1

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

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

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

Answer 2

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

聚合框架非常漂亮，它有许多很酷的运算符，可让您“即时”计算这个缺失的年龄字段。

我要将聚合的每个步骤存储在一个变量中，以便更容易看到发生了什么：

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
}
> 

这不简单的原因是目前聚合框架不支持直接减去日期。请投票给https://jira.mongodb.org/browse/SERVER-6239，它是下一个主要版本的目标——一旦实施，它应该允许直接减去日期（尽管您仍然需要将其转换为适当的粒度，在这种情况下可能是几年）。