PyMongo通过多个键进行分组

Question

使用PyMongo，按一个键分组似乎可以：

results = collection.group(key={"scan_status":0}, condition={'date': {'$gte': startdate}}, initial={"count": 0}, reduce=reducer)

结果：

{u'count': 215339.0, u'scan_status': u'PENDING'} {u'count': 617263.0, u'scan_status': u'DONE'}

但当我尝试按多个键分组时，我会遇到一个例外：

results = collection.group(key={"scan_status":0,"date":0}, condition={'date': {'$gte': startdate}}, initial={"count": 0}, reduce=reducer)

如何正确地按多个字段分组？

Answer 1

如果您试图计数两个以上的键，那么虽然可以使用.group()，但更好的选择是通过.aggregate()。

这使用"本机代码运算符"，而不是.group()所需的JavaScript解释代码来执行与您试图实现的基本"分组"操作相同的操作。

特别是$group管道运营商：

result = collection.aggregate([
    # Matchn the documents possible
    { "$match": { "date": { "$gte": startdate } } },
    # Group the documents and "count" via $sum on the values
    { "$group": {
        "_id": {
            "scan_status": "$scan_status",
            "date": "$date"
        },
        "count": { "$sum": 1 }
    }}
])

事实上，你可能想要一些能将"日期"缩短为一个不同时期的东西。如：

result = collection.aggregate([
    # Matchn the documents possible
    { "$match": { "date": { "$gte": startdate } } },
    # Group the documents and "count" via $sum on the values
    { "$group": {
        "_id": {
            "scan_status": "$scan_status",
            "date": {
                "year": { "$year": "$date" },
                "month": { "$month" "$date" },
                "day": { "$dayOfMonth": "$date" }
            }
        },
        "count": { "$sum": 1 }
    }}
])

使用日期聚合运算符，如下所示。

或者可能有基本的"日期数学"：

import datetime
from datetime import date
result = collection.aggregate([
    # Matchn the documents possible
    { "$match": { "date": { "$gte": startdate } } },
    # Group the documents and "count" via $sum on the values
    # use "epoch" "1970-01-01" as a base to convert to integer
    { "$group": {
        "_id": {
            "scan_status": "$scan_status",
            "date": {
                "$subtract": [
                    { "$subtract": [ "$date", date.fromtimestamp(0) ] },
                    { "$mod": [
                        { "$subtract": [ "$date", date.fromtimestamp(0) ] },
                        1000 * 60 * 60 * 24
                    ]}
                ]
            }
        },
        "count": { "$sum": 1 }
    }}
])

它将返回"epoch"时间的整数值，而不是compisite值对象。

但所有这些选项都比.group()更好，因为它们使用本机编码的例程，并且执行操作的速度比您需要提供的JavaScript代码快得多。