• 如何使用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]])
  • 相关阅读:
    POJ 2155 Matrix(二维树状数组)
    HDU 1280 前m大的数
    HDU 3183 A Magic Lamp(二维RMQ)
    HDU 3743 Frosh Week(归并排序求逆序数)
    POJ 2299 Ultra-QuickSort ( 归并排序 + 求逆序数 )
    HDU 1166 敌兵布阵(树状数组)
    HDU 2846 Repository(字典树)
    HDU 1896 Stones(优先队列)
    HDU 4393 Throw nails(优先队列)
    进程池
  • 原文地址:https://www.cnblogs.com/lhuan/p/9958617.html
Copyright © 2020-2023  润新知