Python使用lambda排序列表或多个键的dicts

时间:2017-06-28 21:40:13

标签: python sorting lambda

这是我的dict列表:

l = [{'a': 2, 'c': 1, 'b': 3}, 
     {'a': 2, 'c': 3, 'b': 1}, 
     {'a': 1, 'c': 2, 'b': 3},
     {'a': 1, 'c': 3, 'b': 2}, 
     {'a': 2, 'c': 5, 'b': 3}]

现在我想按用户提供的键和顺序对列表进行排序。例如:

keys = ['a', 'c', 'b']
orders = [1, -1, 1]

我尝试在sort()方法中使用lambda,但它以一种奇怪的方式失败了:

>>> l.sort(key=lambda x: (order * x[key] for (key, order) in zip(keys, orders)))
>>> l
[{'a': 2, 'c': 5, 'b': 3},
 {'a': 1, 'c': 3, 'b': 2},
 {'a': 1, 'c': 2, 'b': 3},
 {'a': 2, 'c': 3, 'b': 1},
 {'a': 2, 'c': 1, 'b': 3}]

任何人都知道如何解决这个问题?

1 个答案:

答案 0 :(得分:5)

你几乎在那里;你的lambda产生生成器表达式,而那些碰巧按它们的内存地址排序(在Python 2中)并在Python 3中产生TypeError: '<' not supported between instances of 'generator' and 'generator'异常。

改为使用列表理解:

l.sort(key=lambda x: [order * x[key] for (key, order) in zip(keys, orders)])

演示:

>>> l = [{'a': 1, 'c': 2, 'b': 3},
...      {'a': 1, 'c': 3, 'b': 2},
...      {'a': 2, 'c': 1, 'b': 3},
...      {'a': 2, 'c': 5, 'b': 3},
...      {'a': 2, 'c': 3, 'b': 1}]
>>> keys = ['a', 'c', 'b']
>>> orders = [1, -1, 1]
>>> l.sort(key=lambda x: [order * x[key] for (key, order) in zip(keys, orders)])
>>> from pprint import pprint
>>> pprint(l)
[{'a': 1, 'b': 2, 'c': 3},
 {'a': 1, 'b': 3, 'c': 2},
 {'a': 2, 'b': 3, 'c': 5},
 {'a': 2, 'b': 1, 'c': 3},
 {'a': 2, 'b': 3, 'c': 1}]