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

lexsort支持对数组按指定行或列的顺序排序;是间接排序,lexsort不修改原数组,返回索引。

(对应lexsort 一维数组的是argsort a.argsort()这么使用就可以;argsort也不修改原数组, 返回索引)

 

默认按最后一行元素有小到大排序, 返回最后一行元素排序后索引所在位置。

设数组a, 返回的索引ind,ind返回的是一维数组

对于一维数组, a[ind]就是排序后的数组。

对于二维数组下面会详细举例。

 

import numpy as np

 

>>> 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]])

 

按最后一列逆序排序

>>>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]])

转载请注明出处: http://www.cnblogs.com/liyuxia713/

  • 1
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值