本文實(shí)例講述了Python實(shí)現(xiàn)二維數(shù)組按照某行或列排序的方法。分享給大家供大家參考,具體如下:
lexsort支持對(duì)數(shù)組按指定行或列的順序排序;是間接排序,lexsort不修改原數(shù)組,返回索引。
(對(duì)應(yīng)lexsort 一維數(shù)組的是argsort a.argsort()這么使用就可以;argsort也不修改原數(shù)組, 返回索引)
默認(rèn)按最后一行元素有小到大排序, 返回最后一行元素排序后索引所在位置。
設(shè)數(shù)組a, 返回的索引ind,ind返回的是一維數(shù)組
對(duì)于一維數(shù)組, a[ind]就是排序后的數(shù)組。
對(duì)于二維數(shù)組下面會(huì)詳細(xì)舉例。
import numpy as np>>> aarray([[ 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)].Tarray([[ 2, 4, 7, 2], [ 5, 1, 9, 35], [ 2, 3, 12, 22]])
按第一行順序排序
>>> a.T[np.lexsort(a[::-1,:])].Tarray([[ 2, 2, 4, 7], [ 5, 35, 1, 9], [ 2, 22, 3, 12]])
希望本文所述對(duì)大家Python程序設(shè)計(jì)有所幫助。
新聞熱點(diǎn)
疑難解答
圖片精選