我有一个 Python 对象列表,我想按对象本身的属性对其进行排序。该列表如下所示:
>>> ut
[<Tag: 128>, <Tag: 2008>, <Tag: <>, <Tag: actionscript>, <Tag: addresses>,
<Tag: aes>, <Tag: ajax> ...]
每个对象都有一个计数:
>>> ut[1].count
1L
我需要按计数降序对列表进行排序。
我已经看到了几种方法,但我正在寻找 Python 中的最佳实践。
# To sort the list in place...
ut.sort(key=lambda x: x.count, reverse=True)
# To return a new list, use the sorted() built-in function...
newlist = sorted(ut, key=lambda x: x.count, reverse=True)
更多关于按键排序的信息。
一种最快的方法是使用operator.attrgetter("count")
. 但是,这可能会在 Python 的预操作员版本上运行,因此最好有一个回退机制。然后,您可能想要执行以下操作:
try: import operator
except ImportError: keyfun= lambda x: x.count # use a lambda if no operator module
else: keyfun= operator.attrgetter("count") # use operator since it's faster than lambda
ut.sort(key=keyfun, reverse=True) # sort in-place
读者应该注意到 key= 方法:
ut.sort(key=lambda x: x.count, reverse=True)
比向对象添加丰富的比较运算符要快很多倍。我很惊讶地读到这个(“Python in a Nutshell”的第 485 页)。您可以通过在这个小程序上运行测试来确认这一点:
#!/usr/bin/env python
import random
class C:
def __init__(self,count):
self.count = count
def __cmp__(self,other):
return cmp(self.count,other.count)
longList = [C(random.random()) for i in xrange(1000000)] #about 6.1 secs
longList2 = longList[:]
longList.sort() #about 52 - 6.1 = 46 secs
longList2.sort(key = lambda c: c.count) #about 9 - 6.1 = 3 secs
我的,非常小的测试显示第一种慢了 10 倍以上,但书上说它通常只慢 5 倍。他们说的原因是由于 python ( timsort ) 中使用的高度优化的排序算法。
尽管如此,.sort(lambda) 比普通的旧 .sort() 更快,这很奇怪。我希望他们能解决这个问题。
面向对象的方法
如果适用,最好将对象排序逻辑作为类的属性,而不是在需要排序的每个实例中合并。
这确保了一致性并消除了对样板代码的需求。
至少,您应该为此指定__eq__
和__lt__
操作。然后只需使用sorted(list_of_objects)
.
class Card(object):
def __init__(self, rank, suit):
self.rank = rank
self.suit = suit
def __eq__(self, other):
return self.rank == other.rank and self.suit == other.suit
def __lt__(self, other):
return self.rank < other.rank
hand = [Card(10, 'H'), Card(2, 'h'), Card(12, 'h'), Card(13, 'h'), Card(14, 'h')]
hand_order = [c.rank for c in hand] # [10, 2, 12, 13, 14]
hand_sorted = sorted(hand)
hand_sorted_order = [c.rank for c in hand_sorted] # [2, 10, 12, 13, 14]
from operator import attrgetter
ut.sort(key = attrgetter('count'), reverse = True)
它看起来很像 Django ORM 模型实例的列表。
为什么不在这样的查询中对它们进行排序:
ut = Tag.objects.order_by('-count')
将丰富的比较运算符添加到对象类中,然后使用列表的 sort() 方法。
在 python 中查看丰富的比较。
更新:虽然这种方法可行,但我认为 Triptych 的解决方案更适合您的情况,因为它更简单。
如果要排序的属性是property,则可以避免导入operator.attrgetter
并改用 property 的fget
方法。
例如,对于Circle
具有属性的类,radius
我们可以按半径对列表进行排序circles
,如下所示:
result = sorted(circles, key=Circle.radius.fget)
这不是最知名的功能,但经常为我节省了导入的一行。