zoukankan      html  css  js  c++  java
  • 如何使用python来对二维数组进行排序

    1、复合排序

    直接用numpy的lexsort就可以

    import numpy as np
    data = np.array([[1,2,3,4,5], [1,2,3,6,7], [2,3,4,5,7], [3,4,5,6,7], [4,5,6,7,8]])
    idex=np.lexsort([-1*data[:,2], data[:,1], data[:,0]])

    #先按第一列升序,再按第二列升序,再按第三列降序
    #注意先按后边的关键词排序

    sorted_data = data[idex, :]

    2、

    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]])
  • 相关阅读:
    HDU4366 Successor 线段树+预处理
    POJ2823 Sliding Window 单调队列
    HDU寻找最大值 递推求连续区间
    UVA846 Steps 二分查找
    HDU3415 Max Sum of MaxKsubsequence 单调队列
    HDU时间挑战 树状数组
    UVA10168 Summation of Four Primes 哥德巴赫猜想
    UESTC我要长高 DP优化
    HDUChess 递推
    HDU4362 Dragon Ball DP+优化
  • 原文地址:https://www.cnblogs.com/lhuan/p/9958617.html
Copyright © 2011-2022 走看看