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,[ 2,5]])

按最后一列逆序排序

>>>a[np.lexsort(-a.T)]
array([[35,2]])

按第一列顺序排序

>>> a[np.lexsort(a[:,::-1].T)]
array([[ 2,5]])

按最后一行顺序排序

>>> a.T[np.lexsort(a)].T
array([[ 2,[ 5,35],22]])

按第一行顺序排序

>>> a.T[np.lexsort(a[::-1,:])].T
array([[ 2,2,7],35,9],22,12]])

更多关于Python相关内容感兴趣的读者可查看本站专题：《Python数组操作技巧总结》、《Python字符串操作技巧汇总》、《Python函数使用技巧总结》、《Python入门与进阶经典教程》及《Python数据结构与算法教程》

希望本文所述对大家Python程序设计有所帮助。