Python排序函式用法

he_jia發表於2020-12-11
Python排序函式

Python排序函式完美體現了Python語言的簡潔性，對於List物件，我們可以直接呼叫sort()函式(這裡稱為"方法"更合適)來進行排序，而對於其他可迭代物件(如set，dict)，我們可以使用更靈活的sorted()函式。

一.List的sort()函式

Python原始碼builtins.py檔案對sort()函式的定義如下

    def sort(self, key=None, reverse=False):
        """ L.sort(key=None, reverse=False) -> None -- stable sort *IN PLACE* """
        pass

可以看出：sort()函式沒有返回值，有兩個引數。key表示的是排依據的函式；reverse是指需不需要反轉列表，預設為False表示的是升序，如果設為True表示的是降序。

sort()用法如下

L = [1, 2, 7, 4, 3]
L.sort()
print(L)
#[1, 2, 3, 4, 7]

我們加入一些引數來看看，如降序排序：

L = [1, 2, 7, 4, 3]
L.sort(reverse=True)
print(L)
#[7, 4, 3, 2, 1]

再比如，我們想通過key來實現降序排序，這裡用到匿名函式lambda：

L = [1, 2, 7, 4, 3]
L.sort(key=lambda x : x*(-1))
print(L)
#[7, 4, 3, 2, 1]

當然，key還有很多用法，我們可以將其設為內建函式或者自定義函式，比如用絕對值函式：

L = [-1, 2, -7, 4, 3]
L.sort(key=abs)
print(L)
#[-1, 2, 3, 4, -7]

Python3對key的還有另外一種支援，就是通過functools模組的cmp_to_key函式，將傳統的cmp函式轉為key。這和lambda函式用法類似，但是對於比較的邏輯比較複雜的情況，這樣的方式會更清晰更便於維護。

from functools import cmp_to_key
L = [-1, 2, -7, 4, 3]
def cmp(a,b):
    if abs(a) < abs(b):
        return -1
    else:
        return 1
L.sort(key=cmp_to_key(cmp))
print(L)

二.sorted()

比起sort()函式只能排序列表，sorted函式更通用一些。官方文件對其有詳細說明：

sorted(iterable, *, key=None, reverse=False)
Return a new sorted list from the items in iterable.

sorted()返回值為List型別。引數列表iterable表示可迭代物件；*表示位置引數就此終結，後面的引數都必須用關鍵字來指定；key與reverse引數用法與sort()完全一致。

sorted()用法如下：

L = [1, 2, 7, 4, 3]
L1 = sorted(L)
print(L1)
#[1, 2, 3, 4, 7]

對於reverse和key的用法就不單獨展示了，可以參考sort()。如果我們想要排序的物件是一個集合，那麼用法與列表排序一樣：

S = {1, 2, 7, 4, 3}
S1= sorted(S)
print(S1)
#[1, 2, 3, 4, 7]

注意這個排序物件是set物件，而結果是一個列表物件(set物件本就沒有順序一說)。如果我們要實現一個複雜一點的排序呢？比如對字典排序，排序依據是鍵值，排序結果是由鍵值對組成的列表：

Dict = {'a':2, 'b':3, 'c':7, 'd':4, 'e':1}
L = sorted(Dict.items(), key=lambda x : x[1])
print(L)
#[('e', 1), ('a', 2), ('b', 3), ('d', 4), ('c', 7)]

最後給一個Leetcode上的題(根據字元出現頻率排序)：

給定一個字串，請將字串裡的字元按照出現的頻率降序排列。
輸入:
"tree"
輸出:
"eert"
解釋:
'e'出現兩次，'r'和't'都只出現一次。因此'e'必須出現在'r'和't'之前。此外，"eetr"也是一個有效的答案。

通過使用sorted函式程式碼就很清晰易讀了：

def frequencySort(s):
    d = {}
    for i in s:
        d[i] = d.get(i,0) + 1
    L = sorted(d.keys(), key=lambda x: d[x], reverse=True)
    s = ''
    for i in L:
        s += i * d[i]
    return s

注意這裡的d.keys()生成的是一個由key組成的可迭代物件，而前面程式碼的Dict.items()生成的是一個由鍵值對物件(這是一個Tuple物件)組成的可迭代物件。

 

相關文章