在字典列表中有其他格式化方法吗

Question

我有一个字典列表：

data = [
{"id": 55, "name": "Ram", "city": "kathmandu"},
{"id": 66, "name": "Hari", "city": "lalitpur"},
{"id": 77, "name": "Sita", "city": "kathmandu"},
{"id": 88, "name": "Geeta", "city": "pokhara"},
{"id": 99, "name": "Shyam", "city": "pokhara"},
{"id": 95, "name": "Parbati", "city": "bhaktapur"}
]

我希望预期输出为：

{'kathmandu': ['Ram', 'Sita'], 'lalitpur': ['Hari'], 'pokhara': ['Shyam', 'Geeta'], 'bhaktapur': ['Parbati']}

我的解决方案是：

for i in data:
i[i.get('city')] = i['name']
del i['name']
del i['id']
del i['city']

new_data = data[::-1]
first_dict = dict((key, val) for k in data for key, val in k.items())
second_dict = dict((key, val) for k in new_data for key, val in k.items())
for key in first_dict:
if key in second_dict:
new_list = [first_dict[key],second_dict[key]]
new_list2 = list(set(new_list))
second_dict[key] = new_list2

c = {**first_dict, **second_dict}
print(c)

这产生了预期的结果。但是，我想知道这个解决方案的任何替代方法。

Answer 1

您可以使用setdefault:

result={}
for di in data:
result.setdefault(di['city'], []).append(di['name'])
>>> result
{'kathmandu': ['Ram', 'Sita'], 'lalitpur': ['Hari'], 'pokhara': ['Geeta', 'Shyam'], 'bhaktapur': ['Parbati']}

Answer 2

您有一个有效的解决方案，但您正在修改dict并完全重塑它。我建议您只将所需的数据放在new_dict中。

看起来像是defaultdict的作业。

data = [
{"id": 55, "name": "Ram", "city": "kathmandu"},
{"id": 66, "name": "Hari", "city": "lalitpur"},
{"id": 77, "name": "Sita", "city": "kathmandu"},
{"id": 88, "name": "Geeta", "city": "pokhara"},
{"id": 99, "name": "Shyam", "city": "pokhara"},
{"id": 95, "name": "Parbati", "city": "bhaktapur"}
]
from collections import defaultdict
new_data = defaultdict(list)
for record in data:
city = record['city']
name = record['name']
new_data[city].append(name)
print(new_data)