在Python中按给定列表对字典进行排序

Question

有没有一种简单的方法可以按给定列表对字典进行排序？

例：

data = OrderedDict([('pi', 'var1'), ('pa', 'var2'), ('pu', 'var3')])
order = ['pu', 'pi', 'pa']

要求订购的字典：

>>> OrderedDict([('pu', 'var3'), ('pi', 'var1'), ('pa', 'var2')])

提前致谢

Answer 1

您可以通过以下生成器理解来做到这一点：

sorted_data = OrderedDict((k, data[k]) for k in order)

Answer 2

使用lambda函数（ lambda t: order.index(t[0]) ）作为OrderedDict的key参数的一种方法是：

In [1]: from collections import OrderedDict
In [2]: order = ['pu', 'pi', 'pa']
In [3]: data = OrderedDict([('pi', 'var1'), ('pa', 'var2'), ('pu', 'var3')])
In [4]: OrderedDict(sorted(data.items(), key=lambda t: order.index(t[0])))
Out[4]: OrderedDict([('pu', 'var3'), ('pi', 'var1'), ('pa', 'var2')])

文档-https: //docs.python.org/2/library/collections.html#ordereddict-examples-and-recipes

Answer 3

>>> import itertools
>>> from collections import OrderedDict

>>> counter = itertools.count()
>>> input = [('pi', 'var1'), ('pa', 'var2'), ('pu', 'var3')]
>>> order = {key: counter.next() for key in ['pu', 'pi', 'pa']}

>>> OrderedDict(sorted(input, key=lambda pair: order[pair[0]]))
OrderedDict([('pu', 'var3'), ('pi', 'var1'), ('pa', 'var2')])