python自定义规律绘制_Python3 sorted() 自定义排序规则

面试题目：给定两个数N，K和N个字符串，输出2*K行，每行为字符串和该字符串出现的次数，对于前K行，按字符串出现次数降序排列，如果出现次数相同，则按字符串的字典序升序输出，后K行，按字符串出现次数升序排列，如果出现次数相同，则按字符串的字典序升序输出。

题解，其实就是对字符串S的一个计数+排序，排序的对象为：（出现次数，字符串），当使用出现次数Count无法判定大小的时候，使用额外的字符串本身的序列作为比较依据。

使用Java还是比较好写的，直接传给list.sort()一个比较器（Comparator）就行了：

class Row {

String string;

Integer count;

// getter and setter

@Override

public String toString() {

return string + " " + count;

}

}

......

list.sort((Row o1, Row o2) -> {

if (o1.getCount() > o2.getCount()) {

return -1;

} else if (o1.getCount().equals(o2.getCount())) {

return o1.getString().compareTo(o2.getString());

} else {

return 1;

}

});

list.stream().limit(k).forEach(System.out::println);

......

但是之前没有太使用过Python里面的这个比较器，之前的有限经验里面使用的比较多的就是字典的排序了：

# 按照键排

t = sorted(d.items(), key=lambda d: d[0])

# 按照值排

t = sorted(d.items(), key=lambda d: d[1])

没有使用到两个以上的排序属性的排序，查阅之后，发现还是同样使用sorted()，有如下代码：

import random

from functools import cmp_to_key

# n = int(input())

# a = list(map(int, input().split()))

n = 10

a = [random.randint(0, n) for i in range(n)]

def comparator01(o1, o2):

if o1[1] > o2[1]:

return -1

elif o1[1] == o2[1]:

if o1[0] > o2[0]:

return 1

elif o1[0] == o2[0]:

return 0

else:

return -1

else:

return 1

def comparator02(o1, o2):

if o1[1] > o2[1]:

return 1

elif o1[1] == o2[1]:

if o1[0] > o2[0]:

return 1

elif o1[0] == o2[0]:

return 0

else:

return -1

else:

return -1

d = {}

for i in a:

if i in d.keys():

d[i] += 1

else:

d[i] = 1

t = sorted(d.items(), key=cmp_to_key(comparator01))

limit = n // 3

for i in range(limit):

print('{} {}'.format(*t[i]))

t = sorted(d.items(), key=cmp_to_key(comparator02))

for i in range(limit):

print('{} {}'.format(*t[i]))

需要自定义比较器的逻辑，之后使用functools里面的cmp_to_key()将比较逻辑进行包装，传递给key参数。