Python如何按一个列表的顺序对另一个字典列表排序过滤

这个场景不需要套用通用列表排序的逻辑，直接按参考列表顺序做匹配收集即可，实现简单且性能更优。

核心思路 #

• 先把待处理的字典列表list_sec转换成以key字段为索引的映射表，把单元素查找的时间复杂度降到O(1)，避免每次匹配都重复遍历整个列表
• 按顺序遍历参考列表list_main，每拿到一个值就去映射表里查询：如果存在对应的字典元素，就加入结果列表；如果不存在直接跳过，自然完成无匹配项的过滤

可直接运行的示例代码 #

list_main = [3, 33, 2]
list_sec = [{'key': 2, 'rocket': 'mark11'}, {'key': 332, 'rocket': 'mark23'}, {'key': 3, 'rocket': 'mark1'} ]

# 构建key到对应字典的映射
key_map = {item['key']: item for item in list_sec}

# 按参考列表顺序收集匹配项，自动过滤无匹配的元素
output = [key_map[target_key] for target_key in list_main if target_key in key_map]

运行后得到的output和预期结果完全一致：

[{'key': 3, 'rocket': 'mark1'}, {'key': 2, 'rocket': 'mark11'}]

补充说明 #

• 如果list_sec中可能存在多个相同key的字典，只需要调整映射构建逻辑，把同key的元素存入列表，最后遍历的时候展开即可，核心匹配逻辑不变
• 这个实现的总时间复杂度是O(n+m)，比自定义排序键的写法（时间复杂度O(m log m)）性能更好，数据量较大时优势更明显
• 最终结果的顺序完全和list_main的元素顺序对齐，list_sec里存在但list_main没有提到的元素（比如示例中key为332的条目）会被自动排除，不需要额外编写过滤逻辑

内容的提问来源于stack exchange，提问作者Anant