美文网首页
理解sort()函数中的key

理解sort()函数中的key

作者: Elvis_zhou | 来源:发表于2018-10-31 16:55 被阅读0次

    最近在看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开头的元祖(键相同则按值排序),于是便完成了优先排序。
    以上!

    相关文章

      网友评论

          本文标题:理解sort()函数中的key

          本文链接:https://www.haomeiwen.com/subject/lmpatqtx.html