【python】二维数组按照某行或某列排序(numpy lexsort)
这里介绍的排序方法主要是针对最后一行和最后一列,如果要指定某一行目前没有很好的方法,有个笨办法可以先把排序的该列(或行)换到最后,排序后再换回来
曾经以为排序只能使用list的sort函数,后来发现array的排序实现方法也有很多。
List
1、一维
2、二维
无论是一维数组还是二维数组都能通过 list 自带的 sort() 函数进行更新,或者直接调用sorted()方法即可。
import numpy as np
a = [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 - 博客园
更多推荐
所有评论(0)