Python中二維數組依照某行或列排序的實作方法

這篇文章主要介紹了Python實現二維數組按照某行或列排序的方法,結合具體實例形式分析了Python使用numpy模組的lexsort方法針對二維數組進行排序的常用操作技巧,需要的朋友可以參考下

本文實例講述了Python實作二維陣列依照某行或列排序的方法。分享給大家供大家參考，如下：

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

以上是Python中二維數組依照某行或列排序的實作方法的詳細內容。更多資訊請關注PHP中文網其他相關文章！