Совокупный ключ / значение из словарей в PyMongo - PullRequest
Новая
       6

Совокупный ключ / значение из словарей в PyMongo

0 голосов
/ 08 ноября 2018

У меня есть коллекция документов, которые выглядят так: 

{
  "id": 45,
  "some_list": [{"x": 1, "y": {"a": 10, "b": 32, "c": 91}}, 
                {"x": 2, "y": {"a": 12, "b": 26, "d": 75, "p": 1}}]
}

или 

{
  "id": 119,
  "some_list": [{"x": 1, "y": {"a": 5}}, 
                {"x": 2, "y": {"b": 60, "p": 8}}]
}

Я хотел бы получить агрегированный результат, который суммирует все словари y. И получите: 

{"a": 27, "b": 118, "c": 91, "d": 75, "p": 9}

# a = 10 + 12 + 5
# b = 32 + 26 + 60
# c = 91
# d = 75
# p = 1 + 8

Вот что я пробовал: 

pipeline = [
    {"$unwind": "$some_list"},
    {"$group": {"_id": "$some_list.y.key", "count": {"$sum": "$some_list.y.value"}}},
]
list(client.db.collection.aggregate(pipeline))

1 Ответ

0 голосов
/ 08 ноября 2018

Вы можете использовать ниже агрегации

Поскольку у вас есть неизвестные ключи, вы должны использовать $objectToArray для преобразования динамических ключей в формат k и v, тогда вы можете легко $group им. 

db.collection.aggregate([
  { "$unwind": "$some_list" },
  { "$addFields": {
    "some_list": { "$objectToArray": "$some_list.y" }
  }},
  { "$unwind": "$some_list" },
  { "$group": {
    "_id": "$some_list.k",
    "count": { "$sum": "$some_list.v" }
  }}
])

выход 

[
  { "_id": "c", "count": 91 },
  { "_id": "d", "count": 75 },
  { "_id": "p", "count": 9 },
  { "_id": "b", "count": 118 },
  { "_id": "a", "count": 27 }
]

И даже если вам нужны данные в паре ключ-значение, вы можете использовать эту 

db.collection.aggregate([
  { "$unwind": "$some_list" },
  { "$addFields": {
    "some_list": { "$objectToArray": "$some_list.y" }
  }},
  { "$unwind": "$some_list" },
  { "$group": {
    "_id": "$some_list.k",
    "count": { "$sum": "$some_list.v" }
  }},
  { "$group": {
    "_id": null,
    "data": { "$push": { "k": "$_id", "v": "$count" }}
  }},
  { "$replaceRoot": { "newRoot": { "$arrayToObject": "$data" }}}
])

выход 

[
  {
    "a": 27,
    "b": 118,
    "c": 91,
    "d": 75,
    "p": 9
  }
]
...