如何根据具有密钥的另一个列表来订购字典列表？

Question

我有一些ID如下:

req_ids=['964','123','534','645','876','222']

我得到了另一个服务器的响应,不是以任何特定的顺序:

res_result = [{'id':'123', 'name':'Sachin'},
              {'id':'534', 'name':'Vipin'},
              {'id':'222', 'name':'Ram'},
              {'id':'645', 'name':'Anoop'},
              {'id':'964', 'name':'Sani'},
              {'id':'876', 'name':'John'}]

我需要获得res_result与req_ids相同的请求顺序,

    [{'id':'964', 'name':'Sani'},
     {'id':'123', 'name':'Sachin'},
     {'id':'534', 'name':'Vipin'},
     {'id':'645', 'name':'Anoop'},
     {'id':'876', 'name':'John'},
     {'id':'222', 'name':'Ram'}]

如果可能的话,我如何使用内置的python函数代替编写我们自己的循环来执行此逻辑？

Answer 1

如果必须排序,将键映射到整数,然后使用该映射进行排序:

id_to_pos = {key: i for i, key in enumerate(req_ids)}
sorted(res_result, key=lambda d: id_to_pos[d['id']])

演示:

>>> from pprint import pprint
>>> req_ids = ['964', '123', '534', '645', '876', '222']
>>> res_result = [{'id':'123', 'name':'Sachin'},
...               {'id':'534', 'name':'Vipin'},
...               {'id':'222', 'name':'Ram'},
...               {'id':'645', 'name':'Anoop'},
...               {'id':'964', 'name':'Sani'},
...               {'id':'876', 'name':'John'}]
>>> id_to_pos = {key: i for i, key in enumerate(req_ids)}
>>> sorted(res_result, key=lambda d: id_to_pos[d['id']])
[{'id': '964', 'name': 'Sani'}, {'id': '123', 'name': 'Sachin'}, {'id': '534', 'name': 'Vipin'}, {'id': '645', 'name': 'Anoop'}, {'id': '876', 'name': 'John'}, {'id': '222', 'name': 'Ram'}]
>>> pprint(_)
[{'id': '964', 'name': 'Sani'},
 {'id': '123', 'name': 'Sachin'},
 {'id': '534', 'name': 'Vipin'},
 {'id': '645', 'name': 'Anoop'},
 {'id': '876', 'name': 'John'},
 {'id': '222', 'name': 'Ram'}]

但是,你可以完全避免排序,你已经有了正确的顺序,你只需要从id键映射到字典(O(N)),然后按正确的顺序拔出字典(仍然是O(N)):

id_to_dict = {d['id']: d for d in res_result}
[id_to_dict[id] for id in req_ids]

演示:

>>> id_to_dict = {d['id']: d for d in res_result}
>>> [id_to_dict[id] for id in req_ids]
[{'id': '964', 'name': 'Sani'}, {'id': '123', 'name': 'Sachin'}, {'id': '534', 'name': 'Vipin'}, {'id': '645', 'name': 'Anoop'}, {'id': '876', 'name': 'John'}, {'id': '222', 'name': 'Ram'}]
>>> pprint(_)
[{'id': '964', 'name': 'Sani'},
 {'id': '123', 'name': 'Sachin'},
 {'id': '534', 'name': 'Vipin'},
 {'id': '645', 'name': 'Anoop'},
 {'id': '876', 'name': 'John'},
 {'id': '222', 'name': 'Ram'}]