【python】二维数组按照某行或某列排序(numpy lexsort) |
您所在的位置:网站首页 › python3排序函数 › 【python】二维数组按照某行或某列排序(numpy lexsort) |
这里介绍的排序方法主要是针对最后一行和最后一列,如果要指定某一行目前没有很好的方法,有个笨办法可以先把排序的该列(或行)换到最后,排序后再换回来 曾经以为排序只能使用list的sort函数,后来发现array的排序实现方法也有很多。 List1、一维 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)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 |