python--自定義排序函數

自定義排序函數

Python內置的 sorted()函數可對list進行排序：

>>>sorted([36, 5, 12, 9, 21])
[5, 9, 12, 21, 36]

但 sorted()也是一個高階函數，它可以接收一個比較函數來實現自定義排序，比較函數的定義是，傳入兩個待比較的元素 x, y，如果 x 應該排在 y 的前面，返回 -1，如果 x 應該排在 y 的后面，返回 1。如果 x 和 y 相等，返回 0。

因此，如果我們要實現倒序排序，只需要編寫一個reversed_cmp函數：

def reversed_cmp(x, y):
    if x > y:
        return -1
    if x < y:
        return 1
    return 0

這樣，調用 sorted() 並傳入 reversed_cmp 就可以實現倒序排序：

>>> sorted([36, 5, 12, 9, 21], reversed_cmp)
[36, 21, 12, 9, 5]

在python3中語法如下（python3中去除了cmp參數，所以這里用到python自帶的一個將cmp函數轉化為key的值的函數）：

from functools import cmp_to_key
def reversed_cmp(x, y):
    if x > y:
        return -1
    if x < y:
        return 1
    return 0
sorted([36, 5, 12, 9, 21], key = cmp_to_key(reversed_cmp))

sorted()也可以對字符串進行排序，字符串默認按照ASCII大小來比較：

>>> sorted(['bob', 'about', 'Zoo', 'Credit'])
['Credit', 'Zoo', 'about', 'bob']

'Zoo'排在'about'之前是因為'Z'的ASCII碼比'a'小。

任務

對字符串排序時，有時候忽略大小寫排序更符合習慣。請利用sorted()高階函數，實現忽略大小寫排序的算法。

輸入：['bob', 'about', 'Zoo', 'Credit']
輸出：['about', 'bob', 'Credit', 'Zoo']

代碼

from functools import cmp_to_key #方法1
def cmp_ignore_case(s1, s2):
    u1 = s1.upper()
    u2 = s2.upper()
    if u1 < u2:
        return -1
    if u1 > u2:
        return 1
    return 0
print (sorted(['bob', 'about', 'Zoo', 'Credit'], key = cmp_to_key(cmp_ignore_case)))

def cmp_ignore_case(t): #方法2
    return t.upper()
print(sorted(['bob', 'about', 'Zoo', 'Credit'], key = cmp_ignore_case))

print(sorted(['bob','about', 'Zoo', 'Credit'], key = lambda x : x.upper())) #方法3

運行結果

['about', 'bob', 'Credit', 'Zoo']