如何使用多个排序顺序实现SortedContainers.SortedListWithKey?

时间:2017-10-30 21:24:10

标签: python python-3.x data-structures sortedlist sortedcontainers

下面的代码会由于" TypeError:一元的坏操作数类型而失败 - :' str'"例外,但它说明了我想要实现的目标。

from sortedcontainers import SortedListWithKey
contacts = SortedListWithKey(key=lambda val:(val.FirstName, -val.LastName))
class Person(object):
    pass

p1 = Person()
p1.FirstName = 'Amy'
p1.LastName = 'Obama'

p2 = Person()
p2.FirstName = 'Bob'
p2.LastName = 'Obama'

p3 = Person()
p3.FirstName = 'Bob'
p3.LastName = 'Trump'

p4 = Person()
p4.FirstName = 'Amy'
p4.LastName = 'Trump'

contacts.add(p1)
contacts.add(p2)
contacts.add(p3)
contacts.add(p4)

for p in contacts:
    print(p.FirstName, p.LastName)

预期的输出是:

Amy Trump
Amy Obama
Bob Trump
Bob Obama

如果Person.LastName是数字类型成员,则上述代码将起作用。

如果我没有实现SortedList而只是一个内置的未排序列表,我可以简单地执行

contacts.sort(key = operator.itemgetter(0))
contacts.sort(key = operator.itemgetter(1), reverse=True)

但我的情况不适用这个选项。

SortedContainers网站: http://www.grantjenks.com/docs/sortedcontainers/index.html

1 个答案:

答案 0 :(得分:0)

一个不优雅且不高效的方法是

contacts = SortedListWithKey(key=lambda val:(val.FirstName, [-ord(x) for x in val.LastName]))