[python學習] 語言基礎—排序函數(sort()、sorted()、argsort()函數)


python的內建排序函數有 sort、sorted兩個。

1、基礎的序列升序排序直接調用sorted()方法即可

1 ls = list([5, 2, 3, 1, 4])
2 new_ls = sorted(ls)
或者使用ls.sort()即可,直接將ls改變
3 print(new_ls)

需要注意:sort()方法僅定義在list中,而sorted()方法對所有的可迭代序列都有效

並且針對任何的可迭代序列,sorted()都是返回一個list,

1 print(sorted({8: 'D', 2: 'B', 3: 'B', 4: 'E', 5: 'A'}))
2 #輸出為:[2, 3, 4, 5, 8]

 

sorted(iterable,cmp,key,reverse) 

2、使用cmp函數排序,cmp是帶兩個參數的比較函數

 

1 list1 = [('david', 90), ('mary',90), ('sara',80),('lily',95)]
2 
3 print(sorted(list1,cmp = lambda x,y: cmp(x[0],y[0])))#按照第一個位置的字母序排序
4 #[('david', 90), ('lily', 95), ('mary', 90), ('sara', 80)]
5 
6 print(sorted(list1,cmp = lambda x,y: cmp(x[1],y[1])))#按照第二個位置的數字序排序
7 
8 #[('sara', 80), ('david', 90), ('mary', 90), ('lily', 95)]

 

 

3、添加key參數,key 是帶一個參數的函數

list.sort()和sorted()函數使用key參數來指定一個函數,此函數將在每個元素比較前被調用

例如通過key指定的函數來忽略字符串的大小寫

1 print(sorted("This is a test string from Andrew".split(), key=str.lower))
2 #輸出為:['a', 'Andrew', 'from', 'is', 'string', 'test', 'This']

 

 4、用reverse排序

 

1 print(sorted(list1,reverse = True))#逆轉
2 
3 #[('sara', 80), ('mary', 90), ('lily', 95), ('david', 90)]

 

 

5、argsort()函數,是numpy庫中的函數

下面的不想寫了,直接復制過來留個檔!

 

argsort函數
argsort函數返回的是數組值從小到大的索引值

Examples
--------
One dimensional array:一維數組

>>> x = np.array([3, 1, 2])
>>> np.argsort(x)
array([1, 2, 0])

Two-dimensional array:二維數組

>>> x = np.array([[0, 3], [2, 2]])
>>> x
array([[0, 3],
[2, 2]])

>>> np.argsort(x, axis=0) #按列排序
array([[0, 1],
[1, 0]])

>>> np.argsort(x, axis=1) #按行排序
array([[0, 1],
[0, 1]])
#######################################

例1:

>>> x = np.array([3, 1, 2])
>>> np.argsort(x) #按升序排列
array([1, 2, 0])
>>> np.argsort(-x) #按降序排列
array([0, 2, 1])
>>> x[np.argsort(x)] #通過索引值排序后的數組
array([1, 2, 3])
>>> x[np.argsort(-x)]
array([3, 2, 1])
另一種方式實現按降序排序:

>>> a = x[np.argsort(x)]
>>> a
array([1, 2, 3])
>>> a[::-1]
array([3, 2, 1])

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM