提供了编程的基础技术教程

zl程序教程

您现在的位置是：首页 >  其他

当前栏目

一日一技：包含元组的列表，对第一个元素升序第二个元素降序

2023-02-19 12:20:38 时间

摄影：产品经理

鱿鱼圈

大家都知道，在Python里面可以使用.sort方法或者sorted函数对各种数据进行排序，例如：

a = [2, 3, 1, 9, 3, 7, 4]
a.sort()
b = [2, 3, 1, 9, 3, 7, 4]
c = sorted(b, reverse=True)

运行效果如下图所示：

我们也知道，Python中的元组是可以比较大小的。它的比较方法就是先对第一个元素比较，如果第一个元素相等，再比较第二个元素。因此对包含元组的列表也可以这样排序，例如：

d = [(6, 5), (1, 3), (4, 7), (6, 4), (1, 8), (7, 9)]
d.sort()

但现在问题来了，如果要对这个列表排序，但是需要对元组中的第一个元素升序，第二个元素降序，应该怎么办呢？

之前有一个很简单的算法题：输入一串字符串，输出出现次数最高的3个字母：

s = 'asgaurghajsfa;oeiwfasdfasdgrwerw'
count = {}
for letter in s:
    count.setdefault(letter, 0)
    count[letter] += 1

result = sorted(count.items(), key=lambda x: x[1], reverse=True)
print('出现频率最高的3个字母分别是：', result[:3])

运行效果如下图所示：

如果你做过这个算法题，那么你肯定知道怎么做。只需要设置key参数就可以了。在第二个元素前面加个负号：

d = [(6, 5), (1, 3), (4, 7), (6, 4), (1, 8), (7, 9)]
e = sorted(d, key=lambda x: (x[0], -x[1]))

运行效果如下图所示：

现在问题来了，如果被排序的列表里面的元组第二项，不是数字，而是字符串怎么办？字符串前面又不能加负号啊。

例如对如下列表进行排序，按元组第一个元素升序，第二个元素降序：

[(6, 'apple'), (1, 'google'), (4, 'future'), (6, 'zero'), (1, 'stand'), (7, 't')]

可能有同学会想到，可以把字符转成Ascii码再来加负号。这个方法对单个字符确实可行，但是当字符串有多个字符时，这样做就显得很麻烦了。

为了解决这个问题，我们需要回想一下，为什么在Python里面，有一些对象可以比大小，另一些对象却不行？为什么字符串可以比大小，数字可以比大小，但是当我自定义一个类，然后生成对象的时候，却不能比大小？

Python里面能比较大小的对象，是因为他们的类里面有一类特殊的魔术方法：__eq__用来判断是否相等，__lt__用来判断是否小于，__gt__来判断是否大于……，例如：

这样一来，如果想要倒序，只需要把__lt__里面的逻辑反过来就可以了：

我们再来写一下排序函数，对于需要倒序的位置使用这个自定义的类包一下：

class Reversor:
    def __init__(self, value):
        self.value = value
    
    def __eq__(self, other):
        return self.value == other.value

    def __lt__(self, other):
        return other.value < self.value


d = [(6, 5), (1, 3), (4, 7), (6, 4), (1, 8), (7, 9)]
e = sorted(d, key=lambda x: (x[0], Reversor(x[1])))

运行效果如下图所示：

并且，这个类对字符串排序依然适用：

END

相关文章