1. 程式人生 > >如何使用python來對二維陣列進行排序

如何使用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]])