按特定键的顺序对字典列表进行排序

Question

我有一个 list 和 dict 是这样的：

data = [{"uid": "a1", "name": "b1"}, 
        {"uid": "a3", "name": "b3"}, 
        {"uid": "a4", "name": "b4"},
        {"uid": "a5", "name": "b5"}]

现在我想按照 uid 字段的顺序对 list 进行排序，如下所示：

uid_order = ['a1', 'a5', 'a3', 'a4']

这意味着输出应该是：

[{"uid": "a1", "name": "b1"}, 
 {"uid": "a5", "name": "b5"},
 {"uid": "a3", "name": "b3"},
 {"uid": "a4", "name": "b4"}]

当然，我可以创建一个新的空 list，然后使用 2 个 for 循环来检查和添加每个元素，但是有什么优雅的方法（或更好的复杂性）来做到这一点？

Answer 1

O(n) 解：

>>> [*map({d['uid']: d for d in data}.get, uid_order)]
[{'uid': 'a1', 'name': 'b1'},
 {'uid': 'a5', 'name': 'b5'},
 {'uid': 'a3', 'name': 'b3'},
 {'uid': 'a4', 'name': 'b4'}]

Answer 2

如果您想避免创建新列表，可以使用 sort 方法对列表进行适当的排序。这将使用具有 O(n log n) 复杂性的 Timsort，比您建议的 O(n^2) 实现要好得多：

uid_order = {k: i for i, k in enumerate(uid_order)}
data.sort(key=uid_order.get)

为了加快查找uid_order，我将其反转为字典（反转是因为列表将索引映射到项目，而您希望项目到索引）。您现在有一个 O(1) 查找，而不是在每次评估密钥时查找 O(n)。

另一种仅使用函数调用来制作字典的方法：

uid_order = dict(map(reversed, enumerate(uid_order)))

另见：Python Sorting HOW TO

Answer 3

使用sorted-函数，您可以指定该函数应该排序的键。您想要访问每个字典的 "uid" 键的值，并且该值在 uid_order 中的索引决定了该字典在排序列表中的索引：

sorted(data, key = lambda x: uid_order.index(x["uid"]))

这段代码的意思是，列表的每个元素都被传递给lambda-函数。结果值用作排序后的键。