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

您所在的位置:网站首页 python3排序函数 【python】二维数组按照某行或某列排序(numpy lexsort)

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

2023-08-14 23:38| 来源: 网络整理| 查看: 265

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

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



【本文地址】


今日新闻


推荐新闻


CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3