Python的灵活-允许嵌套定义函数并在内层函数的层级上直接调用内层函数

Posted 2022-06-25 昊虹图像算法

考虑如下场景：
list对象的sort()方法的第一参数key要求的函数只有一个输入参数，如果我们想有多个输入参数，怎么办呢？
这个时候我们就可以利用Python函数的嵌套定义来实现。

先上示例代码，然后再做解释吧。

示例代码如下：

def sort_priority(p_list, group):
    def helper(x):
        if x in group:
            return [0, x]
        else:
            return [1, x]
    p_list.sort(key=helper)


list1 = [1, 5, 3, 9, 7, 4, 2, 8, 6]
group1 = [7, 9]
sort_priority(list1, group1)

运行结果如下：

上面这段代码定义了一个函数sort_priority()，其功能是对列表p_list进行排序，排序的时候把group里的元素排在前面。

怎么实现的呢？
主要是利用list对象的sort()方法的第一参数key实现。关于这个参数的详细介绍见博文 https://blog.csdn.net/wenhao_ir/article/details/125406092
这个参数key调用的函数只能有一个参数，所以我们需要在函数sort_priority()的内部再嵌套定义一个函数helper()，函数helper()的输入参数只有一个，就是列表p_list的每一个元素。

有了这个函数helper()，我们便可以在函数sort_priority()中调用list对象的sort()方法了，所以语句

p_list.sort(key=helper)

和函数helper()是处于同一个缩进量层级下的。

函数helper()实际上对列表p_list中的每一个元素进行了分类，即如果元素值为7或9，则返回列表[0,7]或[0,9]，如果元素值不为7或9，则返回列表[0,x]。那么最终相当于对下面这些列表元素进行排序：

[1,1]
[1,5]
[1,3]
[0,9]
[0,7]
[1,4]
[1,2]
[1,8]
[1,6]

结合博文：
https://blog.csdn.net/wenhao_ir/article/details/125406092
中关于列表元素排序的方法，即先排各列表元素中的第0个元素，再排各列表元素中的第1个元素的规则，上面的列表元素会排成下面这样的顺序：

[0,7]
[0,9]
[1,1]
[1,2]
[1,3]
[1,4]
[1,5]
[1,6]
[1,8]

所以待排顺序的中的list1最终排序为：
[7, 9, 1, 2, 3, 4, 5, 6, 8]

总结：通过Python函数嵌套定义，再结合list对象的sort()方法，我们实现了带额外参数的列表排序。