Question

我在mongodb中有一个数据：

{'word': 'good', 'info': [{'tbl_id': 'd1', 'term_freq': 2}, {'tbl_id': 'd2', 'term_freq': 56}, {'tbl_id': 'd3', 'term_freq': 3}]}
{'word': 'spark', 'info': [{'tbl_id': 'd1', 'term_freq': 6}, {'tbl_id': 'd3', 'term_freq': 11}, {'tbl_id': 'd4', 'term_freq': 10}]}
{'word': 'good', 'info': [{'tbl_id': 'd4', 'term_freq': 12}, {'tbl_id': 'd5', 'term_freq': 8}, {'tbl_id': 'd8', 'term_freq': 7}]}
{'word': 'spark', 'info': [{'tbl_id': 'd5', 'term_freq': 6}, {'tbl_id': 'd6', 'term_freq': 11}, {'tbl_id': 'd7', 'term_freq': 10}]}

并且我想减少相同的单词，因此信息应为完整列表。怎么做？

预期输出：

{'word': 'good',
 'info': [{'tbl_id': 'd1', 'term_freq': 2}, {'tbl_id': 'd2', 'term_freq': 56}, {'tbl_id': 'd3', 'term_freq': 3},
          {'tbl_id': 'd4', 'term_freq': 12}, {'tbl_id': 'd5', 'term_freq': 8}, {'tbl_id': 'd8', 'term_freq': 7}]}
{'word': 'spark',
 'info': [{'tbl_id': 'd1', 'term_freq': 6}, {'tbl_id': 'd3', 'term_freq': 11}, {'tbl_id': 'd4', 'term_freq': 10},
          {'tbl_id': 'd5', 'term_freq': 6}, {'tbl_id': 'd6', 'term_freq': 11}, {'tbl_id': 'd7', 'term_freq': 10}]}

Answer 1

下面的汇总查询将为您提供按单词分组的综合信息列表

db.collection.aggregate([{'$unwind':'$info'},{'$group':{'_id':'$word','info':{'$push':'$info'}}}])

输出：

{
    "_id" : "spark",
    "info" : [
        {
            "tbl_id" : "d1",
            "term_freq" : 6
        },
        {
            "tbl_id" : "d3",
            "term_freq" : 11
        },
        {
            "tbl_id" : "d4",
            "term_freq" : 10
        },
        {
            "tbl_id" : "d5",
            "term_freq" : 6
        },
        {
            "tbl_id" : "d6",
            "term_freq" : 11
        },
        {
            "tbl_id" : "d7",
            "term_freq" : 10
        }
    ]
}
{
    "_id" : "good",
    "info" : [
        {
            "tbl_id" : "d1",
            "term_freq" : 2
        },
        {
            "tbl_id" : "d2",
            "term_freq" : 56
        },
        {
            "tbl_id" : "d3",
            "term_freq" : 3
        },
        {
            "tbl_id" : "d4",
            "term_freq" : 12
        },
        {
            "tbl_id" : "d5",
            "term_freq" : 8
        },
        {
            "tbl_id" : "d8",
            "term_freq" : 7
        }
    ]
}

如何在pymongo中使用group关键字连接数组

1 个答案: