python - 根据另一个字典对字典列表进行排序

Question

我有这两个字典列表:

list_1 = [{'document_id': '5'}, {'document_id': '2'}, {'document_id': '4'}]

list_2 = [{'document_id': '4', 'page_id': '1'}, {'document_id': '5', 'page_id': '2'}, {'document_id': '5', 'page_id': '1'}]

list_1 是 list_2 的超集(就 document_id 而言)(如果这有帮助的话)。

我想要

1. 根据list_1对list_2进行排序
2. 然后根据每个不同 document_id 的 page_id 对 list_2 进行排序

因此，我想最终得到这个:

new_list_2 = [{'document_id': '5', 'page_id': '1'}, {'document_id': '5', 'page_id': '2'}, {'document_id': '4', 'page_id': '1'}]

我怎样才能有效地做到这一点？

Answer 1

通过字典理解创建 list_1 中排序的映射。

docs = {doc.get('document_id'): n for n, doc in enumerate(list_1)}
# >>> docs
# {'5': 0, '2': 1, '4': 2}

然后使用以 page_id 作为排序键的排序。

list_2.sort(key=lambda r: (docs[r['document_id']], r['page_id']))

>>> list_2
[{'document_id': '5', 'page_id': '1'},
 {'document_id': '5', 'page_id': '2'},
 {'document_id': '4', 'page_id': '1'}]