本篇文章主要講:
- 自定義規則排序
- 多字段排序
開講之前,先講一些簡單sorted()
或者sort()
,兩者返回值不同!大家自行學習,不是本文的重點!
sorted([5, 2, 3, 1, 4]) # 輸出:[1, 2, 3, 4, 5]
另一個呢
a = [5, 2, 3, 1, 4]
a.sort()
a
# 輸出:[1, 2, 3, 4, 5]
Key 的使用
比如,如何把數組按照長度大小進行排序,只需設置key
等於一個函數即可
sorted(["adf", "Adfdg", "aDfgbdfafg", "Gahgfh"], key = len) #輸出 ['adf', 'adfdg', 'gahgfh', 'adfgbdfafg']
按照小寫字母排序
sorted(["adf", "Adfdg", "aDfgbdfafg", "Gahgfh"], key = str.lower) #輸出 ['adf', 'Adfdg', 'aDfgbdfafg', 'Gahgfh']
上面都是自帶函數,下面就開始學習。
自定義規則排序
直接上題目,給定一組非負整數,重新排列它們的順序使之組成一個最大的整數。
輸入: [3,30,34,5,9]
輸出: 9534330
這道題就是自定義排序,它有自己的排序規則,規則是 x + y > y + x
,讓x
排在y
前面
這里x,y
表示任意兩個數的字符串。
這里我提供兩種辦法
第一種,使用 cmp_to_key
,因為Python3
取消了cmp
def largestNumber(nums):
from functools import cmp_to_key
def helper(x, y):
if x + y > y + x:
return -1
elif x + y < y + x:
return 1
else:
return 0
return "".join(sorted(map(str, nums), key=cmp_to_key(helper))).lstrip("0") or "0"
執行語句 largestNumber([3,30,34,5,9])
輸出:'9534330'
第二種,使用類的魔法方法
def largestNumber( nums) -> str:
class large_num(str):
def __lt__(self, other):
return self + other > other + self
return "".join(sorted(map(str, nums), key=large_num)).lstrip("0") or "0"
得到結果也是正確的!接下來看看
多字段排序
給你數組a = [[2,3],[4,1],(2,8),(2,1),(3,4)]
,按照第一個數從小到大排序,如果第一個數相同,就按第二個數從大到小。
a = [[2,3],[4,1],(2,8),(2,1),(3,4)]
sorted(a, key = lambda x: (x[0], -x[1])) # 輸出:[(2, 8), [2, 3], (2, 1), (3, 4), [4, 1]]
其實掌握自定義規則排序,多字段排序也就會了,因為它是自定義排序的子集!
如果大家對python排序還有更深理解,歡迎留言,一起討論!
參考資料:
【1】:https://docs.python.org/3/howto/sorting.html
【2】:https://blog.csdn.net/hanshileiai/article/details/44241385