我有一个以适当的 oder 排序的 id 列表:
ids = [1, 2, 4, 6, 5, 0, 3]
我还有一个字典列表,以某种随机方式排序:
rez = [{'val': 7, 'id': 1}, {'val': 8, 'id': 2}, {'val': 2, 'id': 3}, {'val': 0, 'id': 4}, {'val': -1, 'id': 5}, {'val': -4, 'id': 6}, {'val': 9, 'id': 0}]
我的目的是以对应于ids
的方式对列表进行排序rez
:
rez = [{'val': 7, 'id': 1}, {'val': 8, 'id': 2}, {'val': 0, 'id': 4}, {'val': -4, 'id': 6}, {'val': -1, 'id': 5}, {'val': 9, 'id': 0}, {'val': 2, 'id': 3}]
我试过了:
rez.sort(key = lambda x: ids.index(x['id']))
然而,这种方式对我来说太慢了,因为 len(ids)> 150K,而且每个字典实际上都有很多键(有些值是字符串)。任何建议如何以最pythonic但仍然最快的方式做到这一点?
sort
,因为ids
指定了结果的整个顺序。您只需要通过其 id 选择正确的元素:
rez_dict = {d['id']:d for d in rez}
rez_ordered = [rez_dict[id] for id in ids]
这给了:
>>> rez_ordered
[{'id': 1, 'val': 7}, {'id': 2, 'val': 8}, {'id': 4, 'val': 0}, {'id': 6, 'val': -4}, {'id': 5, 'val': -1}, {'id': 0, 'val': 9}, {'id': 3, 'val': 2}]
这应该比排序更快,因为它平均可以在线性时间内完成,而排序是 O(nlogn)。
请注意,这假设每个 id 将有一个条目,如您的示例所示。
我认为你走在正确的轨道上。如果您需要加快速度,因为列表太长并且具有二次复杂性,您可以先将列表转换为字典,将 id 映射到它们各自的索引。
indices = {id_: pos for pos, id_ in enumerate(ids)}
rez.sort(key = lambda x: indices[x['id']])
这样,indices
{0: 5, 1: 0, 2: 1, 3: 6, 4: 2, 5: 4, 6: 3}
,rez
[{'id': 1, 'val': 7},
{'id': 2, 'val': 8},
{'id': 4, 'val': 0},
{'id': 6, 'val': -4},
{'id': 5, 'val': -1},
{'id': 0, 'val': 9},
{'id': 3, 'val': 2}]