Question

我有一本字典

a = {'ground': obj1, 'floor 1': obj2, 'basement': obj3}

我有一个清单。

a_list = ['floor 1', 'ground', 'basement']

我想根据列表使用键来对字典a进行排序。有可能吗？

即：

sort(a).based_on(a_list) #this is wrong. But I want something like this.

输出不必是另一个字典，我不介意将字典转换为元组然后对它们进行排序。

Answer 1

天真的方式，使用sorted() function和自定义排序键（为(key, value)生成的每个dict.items())对调用）对（键，值）元组列表进行排序：< / p>

sorted(a.items(), key=lambda pair: a_list.index(pair[0]))

更快的方法是首先创建索引图：

index_map = {v: i for i, v in enumerate(a_list)}
sorted(a.items(), key=lambda pair: index_map[pair[0]])

这更快，因为index_map中的字典查找需要O（1）常量时间，而a_list.index()调用每次都必须扫描列表，因此需要O（N）线性时间。由于为字典中的每个键值对调用该扫描，因此天真排序选项需要O（N ^ 2）二次时间，而使用映射可保持排序有效（O（N log N），线性时间）。 / p>

两者均假设a_list包含a中找到的所有键。但是，如果是这种情况，那么您也可以反转查找并按按顺序检索密钥：

[(key, a[key]) for key in a_list if key in a]

需要O（N）线性时间，并允许a_list中a中不存在的额外键。

明确：O（N）＆gt; O（N log N）> O（N ^ 2），见this cheat sheet for reference。

演示：

>>> a = {'ground': 'obj1', 'floor 1': 'obj2', 'basement': 'obj3'}
>>> a_list = ('floor 1', 'ground', 'basement')
>>> sorted(a.items(), key=lambda pair: a_list.index(pair[0]))
[('floor 1', 'obj2'), ('ground', 'obj1'), ('basement', 'obj3')]
>>> index_map = {v: i for i, v in enumerate(a_list)}
>>> sorted(a.items(), key=lambda pair: index_map[pair[0]])
[('floor 1', 'obj2'), ('ground', 'obj1'), ('basement', 'obj3')]
>>> [(key, a[key]) for key in a_list if key in a]
[('floor 1', 'obj2'), ('ground', 'obj1'), ('basement', 'obj3')]

Answer 2

您可以按照列表提供的键的顺序检索值，并从键值对中创建一个新列表。

示例：

d = a      # dictionary containing key-value pairs that are to be ordered
l = a_list # list of keys that represent the order for the dictionary
# retrieve the values in order and build a list of ordered key-value pairs
ordered_dict_items = [(k,d[k]) for k in l]

如何基于python中的列表对字典进行排序

2 个答案: