python - 解析和创建嵌套字典

Question

我想创建一个包含字典嵌套结构的字典，如下所示:

{
  "Jaque": {
    "ES": {
      "Madrid": [
        {
          "experience": 9
        }
      ]
    },
    "FR": {
      "Lyon": [
        {
          "experience": 11.4
        }
      ],
      "Paris": [
        {
          "experience": 20
        }
      ]
    }
  },
  "James": {
    "UK": {
      "London": [
        {
          "experience": 10.9
        }
      ]
    }
  },
  "Henry": {
    "UK": {
      "London": [
        {
          "experience": 15
        }
      ]
    }
  },
  "Joe": {
    "US": {
      "Boston": [
        {
          "experience": 100
        }
      ]
    }
  }
}
}

我的输入是这种格式的字典列表:

c = [{
    "country": "US",
    "city": "Boston",
    "name": "Joe",
    "experience": 100
  },
  {
    "country": "FR",
    "city": "Paris",
    "name": "Jaque",
    "experience": 20
  },
  {
    "country": "FR",
    "city": "Lyon",
    "name": "Jaque",
    "experience": 11.4
  },
  {
    "country": "ES",
    "city": "Madrid",
    "name": "Jaque",
    "experience": 9
  },
  {
    "country": "UK",
    "city": "London",
    "name": "Henry",
    "experience": 15
  },
  {
    "country": "UK",
    "city": "London",
    "name": "James",
    "experience": 10.9
  }
]

我的第一种方法是逐步创建嵌套字典:

dd = dict.fromkeys([i.get("name") for i in c],defaultdict(dict))

#will create
# dd = {'Joe': defaultdict(<class 'dict'>, {}), 'Jaque': defaultdict(<class 'dict'>, {}), 'James': defaultdict(<class 'dict'>, {}), 'Henry': defaultdict(<class 'dict'>, {})}

for i in dd:
     for j in c:
         #verify if name from d is in dict j 
         if i in j.values():
             dd[i]=dict(zip([a.get("country") for a in c if i in a.values() ],[b.get("city") for b in c if i in b.values() ]))

# dd will become 
#{'Joe': {'US': 'Boston'}, 'Jaque': {'FR': 'Lyon', 'ES': 'Madrid'}, 'Henry': {'UK': 'London'}, 'James': {'UK': 'London'}}

现在我想不出一种方法来创建/更新 dict dd 的嵌套结构。有没有更动态的方式来创建字典？谢谢

Answer 1

您可以使用 itertools.groupby 以与预期输出类似的方式组织列表，然后循环转换为字典。

from itertools import groupby
from operator import itemgetter

data = [{"country": "US", "city": "Boston", "name": "Joe", "experience": 100 }, {"country": "FR", "city": "Paris", "name": "Jaque", "experience": 20 }, {"country": "FR", "city": "Lyon", "name": "Jaque", "experience": 11.4 }, {"country": "ES", "city": "Madrid", "name": "Jaque", "experience": 9 }, {"country": "UK", "city": "London", "name": "Henry", "experience": 15 }, {"country": "UK", "city": "London", "name": "James", "experience": 10.9 } ]

result = {}
for key, values in groupby(sorted(data, key=itemgetter('name')), key=itemgetter('name')):
    result[key] = {
        v['country']: {v['city']: [{'experience': v['experience']}]} for v in values
        }

print(result)
# {'Henry': {'UK': {'London': [{'experience': 15}]}}, 'James': {'UK': {'London': [{'experience': 10.9}]}}, 'Jaque': {'FR': {'Lyon': [{'experience': 11.4}]}, 'ES': {'Madrid': [{'experience': 9}]}}, 'Joe': {'US': {'Boston': [{'experience': 100}]}}}