这里介绍的排序方法主要是针对最后一行和最后一列,如果要指定某一行目前没有很好的方法,有个笨办法可以先把排序的该列(或行)换到最后,排序后再换回来
曾经以为排序只能使用list的sort函数,后来发现array的排序实现方法也有很多。
List
1、一维
2、二维
无论是一维数组还是二维数组都能通过 list 自带的 sort() 函数进行更新,或者直接调用sorted()方法即可。
import numpy as npa = [3,4,5,1,2,3]b = [[3,4,5],[2,3,4]]c = [[3,4,5],[3,3,4]]a.sort() b.sort()c.sort()print(a)#直接调用sorted()方法,print(sorted(a)) print(b)print(c)
Array
1、一维数组
1)argsort()
argsort()是返回数据排序由小到大排序后的索引,可使用数组自带的a.argsort(),或者使用np.argsort(a),因此排序数组可用以下写法:
>>>a = a[np.argsort(a)] 等价于a=a[a.argsort()] #按升序排列
>>>a = a[np.argsort(a)] #按降序排列
另一种降序排列的方法:(此方法验证不可用)
>>> a[::-1]这是翻转矩阵,不是降序排列,a[::-1] = [3 2 1 5 4 3]
注意和a[:-1]的区别a[:-1]表示从元组中切片,默认从第一个元素开始,到倒数第一个元素前面的那个元素为止a[:-1] = [3 2 1 5 4]
a = np.array([3,4,5,1,2,3])print(a.argsort())# 同np.argsort(a)
得到数组的排序索引
根据索引,我们采用a[a.argsort()]即可得到
a = np.array([3,4,5,1,2,3])print(a[a.argsort()])
2二维数组
1)argsort()只能用来查看排序情况
>>> 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]])
2)lexsort()
lexsort,该函数的作用是找到排序的排序下标,默认按最后一行元素有小到大排序,支持对数组按指定行或列的顺序排序;是间接排序,lexsort不修改原数组,返回索引。
(对应lexsort一维数组使用的是argsort,a.argsort()这么使用就可以;argsort也不修改原数组, 返回索引)
>>> a
array([[ 2,7,4,2],
[35,9,1,5],
[22, 12,3,2]])
按最后一列顺序排序
>>> a[np.lexsort(a.T)]
array([[22, 12,3,2],
[ 2,7,4,2],
[35,9,1,5]])
lexsort(a.T)该函数默认是按行排序,因此需要转置才能拿到按列排序的序号
按最后一列逆序排序
>>>a[np.lexsort(-a.T)]
array([[35,9,1,5],
[ 2,7,4,2],
[22, 12,3,2]])
按第一列顺序排序
>>>a[np.lexsort(a[:,::-1].T)]
array([[ 2,7,4,2],
[22, 12,3,2],
[35,9,1,5]])
按最后一行顺序排序
>>> a.T[np.lexsort(a)].T
array([[ 2,4,7,2],
[ 5,1,9, 35],
[ 2,3, 12, 22]])
按第一行顺序排序
>>> a.T[np.lexsort(a[::-1,:])].T
array([[ 2,2,4,7],
[ 5, 35,1,9],
[ 2, 22,3, 12]])
参考博客:
python 按二维数组的某行或列排序 (numpy lexsort) - liyuxia713 - 博客园
[python学习] 语言基础—排序函数(sort()、sorted()、argsort()函数) - lee-yang - 博客园