python - 将多个键/值关联到 defaultdict(list)
问题描述
我有带有记录列表的 json_data 对象。我需要列表中的所有“id”,并且我想以某种方式将“房间”和“名称”键中的值与 id 列表相关联。我需要列表中的所有 ID 的原因是因为我正在使用一个 API,它将 ID 作为批量处理的列表。我还需要将 'name' 和 'room' 键的值关联到 id 列表。我可以使用 defaultdict(list) 将“房间”关联到列表,但不确定如何也关联“名称”键的值。我该怎么做?
json_data = {
"records": [
{
"id": 123,
"name": "John",
"room": "32B"
},
{
"id": 456,
"name": "Mary",
"room": "32B"
},
{
"id": 789,
"name": "Gary",
"room": "16F"
}]
}
data_list = json_data['records']
my_default_dict = defaultdict(list)
for data in data_list:
my_default_dict[data['room']].append(data['id'])
# This gives {'32B': [123, 456], '16F': [789]}
解决方案
为什么不将您的数据转换为 ID 映射?然后,您可以从地图中提取 ID 列表作为键。
records = json_data['records']
# Warning! This destructively iterates over json_data.
json_map = dict(zip((d.pop('id') for d in records), records))
json_map
# {123: {'name': 'John', 'room': '32B'},
# 456: {'name': 'Mary', 'room': '32B'},
# 789: {'name': 'Gary', 'room': '16F'}}
id_list = list(json_map)
# [123, 456, 789]