1. 程式人生 > >Python中的排序函式argsort,sort,sorted用法

Python中的排序函式argsort,sort,sorted用法

argsort()函式

<span style="font-size:18px;">>>> 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])   </span>

sort()函式和sorted()函式

sort函式是list列表中的函式,而sorted可以對list或者iterator進行排序

1、用sort函式對列表排序時會影響列表本身,而sorted不會

>>> a = [1,2,1,4,3,5]
	>>> a.sort()
	>>> a
	[1, 1, 2, 3, 4, 5]
>>> a = [1,2,1,4,3,5]
	>>> sorted(a)
	[1, 1, 2, 3, 4, 5]
	>>> a
	[1, 2, 1, 4, 3, 5]

2、
sorted(iterable,cmp,key,reverse) 引數:iterable可以是list或者iterator cmp是帶兩個引數的比較函式; key是帶一個引數的函式; reverse為False或者True; (1)用cmp函式排序
<span style="white-space:pre">	</span>>>> list1 = [('david', 90), ('mary',90), ('sara',80),('lily',95)]
	>>> sorted(list1,cmp = lambda x,y: cmp(x[0],y[0]))
	[('david', 90), ('lily', 95), ('mary', 90), ('sara', 80)]
	>>> sorted(list1,cmp = lambda x,y: cmp(x[1],y[1]))
	[('sara', 80), ('david', 90), ('mary', 90), ('lily', 95)]
(2)用key函式排序
<span style="white-space:pre">	</span>>>> list1 = [('david', 90), ('mary',90), ('sara',80),('lily',95)]
	>>> sorted(list1,key = lambda list1: list1[0])
	[('david', 90), ('lily', 95), ('mary', 90), ('sara', 80)]
	>>> sorted(list1,key = lambda list1: list1[1])
	[('sara', 80), ('david', 90), ('mary', 90), ('lily', 95)]

(3)用reverse排序
<span style="white-space:pre">	</span>>>> sorted(list1,reverse = True)
	[('sara', 80), ('mary', 90), ('lily', 95), ('david', 90)]
(4)用operator.itemgetter函式排序
>>> from operator import itemgetter
>>> sorted(list1, key=itemgetter(1))
[('sara', 80), ('david', 90), ('mary', 90), ('lily', 95)]
>>> sorted(list1, key=itemgetter(0))
[('david', 90), ('lily', 95), ('mary', 90), ('sara', 80)]

介紹operator.itemgetter函式
<span style="white-space:pre">	</span>>>> import operator
	>>> a = [1,2,3]
	>>> b = operator.itemgetter(0)
	>>> b(a)
	1

operator.itemgetter函式獲取的不是值,而是定義了一個函式。

(5)多級排序
>>> sorted(list1, key=itemgetter(0,1))
[('david', 90), ('lily', 95), ('mary', 90), ('sara', 80)]