Python实现二维数组按照某行或列排序的方法【numpy lexsort】
这里有新鲜出炉的Python3 Cookbook中文版,程序狗速度看过来!
Python编程语言
Python 是一种面向对象、解释型计算机程序设计语言,由Guido van Rossum于1989年底发明,第一个公开发行版发行于1991年。Python语法简洁而清晰,具有丰富和强大的类库。它常被昵称为胶水语言,它能够把用其他语言制作的各种模块(尤其是C/C++)很轻松地联结在一起。
这篇文章主要介绍了Python实现二维数组按照某行或列排序的方法,结合具体实例形式分析了Python使用numpy模块的lexsort方法针对二维数组进行排序的常用操作技巧,需要的朋友可以参考下
本文实例讲述了Python实现二维数组按照某行或列排序的方法。分享给大家供大家参考,具体如下:
lexsort支持对数组按指定行或列的顺序排序;是间接排序,lexsort不修改原数组,返回索引。
(对应lexsort 一维数组的是argsort a.argsort()这么使用就可以;argsort也不修改原数组, 返回索引)
默认按最后一行元素有小到大排序, 返回最后一行元素排序后索引所在位置。
设数组a, 返回的索引ind,ind返回的是一维数组
对于一维数组, a[ind]就是排序后的数组。
对于二维数组下面会详细举例。
importnumpyasnp
>>>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程序设计有所帮助。
来源: http://www.phperz.com/article/17/1014/349574.html