考虑如下场景:
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()方法,我们实现了带额外参数的列表排序。