python列表排序
簡單記一下python中List的sort方法(或者sorted內建函數)的用法。
關鍵字:
python列表排序 python字典排序 sorted
List的元素可以是各種東西,字符串,字典,自己定義的類等。
sorted函數用法如下:
sorted(data,?cmp=None,?key=None,?reverse=False)
其中,data是待排序數據,可以使List或者iterator, cmp和key都是函數,這兩個函數作用與data的元素上產生一個結果,sorted方法根據這個結果來排序。
cmp(e1, e2) 是帶兩個參數的比較函數, 返回值: 負數: e1 < e2, 0: e1 == e2, 正數: e1 > e2. 默認為 None, 即用內建的比較函數.
key 是帶一個參數的函數, 用來為每個元素提取比較值. 默認為 None, 即直接比較每個元素.
通常, key 和 reverse 比 cmp 快很多, 因為對每個元素它們只處理一次; 而 cmp 會處理多次.
通過例子來說明sorted的用法:
1. 對由tuple組成的List排序
>>>?students?=?[('john','A',15),?('jane','B',12),?('dave','B',10),]
用key函數排序(lambda的用法見 注釋1)
>>>?sorted(students,?key=lambdastudent?:?student[2])#?sort?by?age
[('dave','B',10),?('jane','B',12),?('john','A',15)]
用cmp函數排序
>>>?sorted(students,?cmp=lambdax,y?:?cmp(x[2],?y[2]))#?sort?by?age
[('dave','B',10),?('jane','B',12),?('john','A',15)]
用 operator 函數來加快速度, 上面排序等價于:(itemgetter的用法見 注釋2)
>>>from operator impor titemgetter,?attrgetter
>>>?sorted(students,?key=itemgetter(2))
用 operator 函數進行多級排序
>>>?sorted(students,?key=itemgetter(1,2))#?sort?by?grade?then?by?age
[('john','A',15),?('dave','B',10),?('jane','B',12)]
2. 對由字典排序
>>>?d?=?{'data1':3,'data2':1,'data3':2,'data4':4}
>>>?sorted(d.iteritems(),?key=itemgetter(1),?reverse=True)
[('data4',4),?('data1',3),?('data3',2),?('data2',1)]
LEETCODE關聯題目
Given a non-empty array of integers, return thekmost frequent elements.
For example,
Given[1,1,1,2,2,3]and k = 2, return[1,2].
Note:
You may assumekis always valid, 1 ≤k≤ number of unique elements.
Your algorithm's time complexitymust bebetter than O(nlogn), wherenis the array's size.