Skip to main content
 首页 » 编程设计

python之优化函数以对元组列表进行排序

2025年05月04日12zlslch

我需要按第一项降序排列元组列表,然后按第二项升序排列元组列表。

为此,我实现了以下功能,但我认为它可能会更快。

>>> compare = lambda a, b: -cmp(b[1], a[1]) if b[0] == a[0] else cmp(b[0], a[0]) 
>>> sorted([(0, 2), (0, 1), (1, 0), (1, 2)], cmp=compare) 
[(1, 0), (1, 2), (0, 1), (0, 2)] 

可以优化吗?查看与内置函数的比较:

>>> timeit.Timer(stmt='sorted([(int(random.getrandbits(4)),int(random.getrandbits(4))) for x in xrange(10)], cmp=compare)', setup='import random; compare=compare = lambda a, b: -cmp(b[1], a[1]) if b[0] == a[0] else cmp(b[0], a[0])').timeit(100000) 
4.0584850867917339 
>>> timeit.Timer(stmt='sorted([(int(random.getrandbits(4)),int(random.getrandbits(4))) for x in xrange(10)])', setup='import random').timeit(100000) 
2.6582965153393161 

请您参考如下方法:

对我来说,使用键而不是比较函数要快一点,而且可以说也更容易阅读:

sorted([(0, 2), (0, 1), (1, 0), (1, 2)], key = lambda x:(-x[0], x[1])) 

这需要 Python 2.4 或更新版本。