【python】二维数组按照某行或某列排序(numpy lexsort)

这里介绍的排序方法主要是针对最后一行和最后一列，如果要指定某一行目前没有很好的方法，有个笨办法可以先把排序的该列（或行）换到最后，排序后再换回来

曾经以为排序只能使用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 - 博客园