最近在看effective-python,第二章函数中提到了优先排序的概念,具体代码如下:
values = [1, 5, 3, 9, 7, 4, 2, 8, 6]
group = [7, 9]
def sort_priority(values, group):
def helper(x):
if x in group:
return (0, x)
return (1, x)
values.sort(key=helper)
sort_priority(values, group)
print(values)
[7, 9, 1, 2, 3, 4, 5, 6, 8]
这代码的目的是group优先于values排序(前提就是values包含group),使用了闭包和sort()函数,很熟悉,正好复习一下。
sort(key=None, reverse=False)
sort()有两个参数,一个是reverse,这个非常简单,就是字面意思,如果设为True则反向排序(godv:???)。另一个就是今天重点要说的,key,通常在字典的排序中可以见到这个参数(一般字典排序用sorted()):
dict = {'b': 1, 'r': 7, 'z': 2}
content = list(dict.items()) # 先转换为列表包含元祖的形式
打印如下:
[('b', 1), ('z', 2), ('r', 7)]
content.sort(key=lambda x:x[0]) # 这里的0就是元祖的第一个元素,也就是按键排序。按值就是把0改成1
打印如下:
[('b', 1), ('c', 7), ('z', 2)]
原理:在排序之前,content里的所有元素都会执行key的函数,这里指的就是lambda函数,计算出值之后,赋值给key(相当于{'b': ('b', 1)}),然后sort()是针对key进行排序,然后再根据这个key对应的值替换到排好序的content里。
看到这里,优先排序的代码就非常清晰了:
helper()函数判断value中哪些元素在group中,并把其设置为以0为第一个元素的元祖,而不在group的元素则设置为以1为第一个元素的元祖,所以在sort()函数排序时,位置靠前的当然时以0开头的元祖(键相同则按值排序),于是便完成了优先排序。
以上!