300字范文,内容丰富有趣,生活中的好帮手!
300字范文 > python二维数组排序_Python实现二维数组按照某行或列排序的方法【numpy lexsort】...

python二维数组排序_Python实现二维数组按照某行或列排序的方法【numpy lexsort】...

时间:2022-09-21 11:16:41

相关推荐

python二维数组排序_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, 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程序设计有所帮助。

如您对本文有疑问或者有任何想说的,请点击进行留言回复,万千网友为您解惑!

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。