1. 程式人生 > >python 按二維陣列的某行或列排序 (numpy lexsort)

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/