考慮如下場景:
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()方法,我們實現了帶額外參數的列表排序。