numpy.lexsort(): 排序

最新推荐文章于 2025-04-21 20:45:45 发布

原创最新推荐文章于 2025-04-21 20:45:45 发布 · 3.5k 阅读

7 ·

CC 4.0 BY-SA版权

文章标签：

#Ubuntu #numpy

Numpy 专栏收录该内容

4 篇文章

订阅专栏

numpy.lexsort()函数用于在指定维度上对数组进行多重排序。它首先按最后一个数组进行排序，对于相同元素，使用前一个数组进行辅助排序，以此类推。文章通过实例详细解释了lexsort的工作机制，并提供了代码验证。

部署运行你感兴趣的模型镜像

官网的定义:

解释:在指定的维度下,对给定的数组进行从小到大排序,最后一个数组为主排序数组,其与数组为辅助排序数组,返回的是主排序数组的索引.

例子:

$a = \left [ 1, 5,1,4,3,4,4\right ]$

$b = \left [ 9, 4, 0, 4, 0, 2, 1 \right ]$

加上索引后的a:

$a = \left [ 1_{0} ,5_{1} , 1_{2}, 4_{3}, 3_{4}, 4_{5}, 4_{6}}}}}} \right ]$

加上索引后的b:

$b = \left [ 9_{0}, 4_{1}, 0_{2}, 4_{3}, 0_{4}, 2_{5}, 1_{6}}}}}}}} \right ]$

numpy.lexsort((b, a)) 函数的含义:

先对数组a按照从小到大的顺序进行排序, 对于数组a中相同的数字,无法进行排序比较.则去数组b中找相同索引所对应的值,并依据他们的大小来对a中相同索引对应的值进行排序,最后返回排好序之后的数组a的索引值.

对a进行初次排序后的值如下,其中3和5的索引能确定,但是1和4现在没法确定,因为当出现相同的数字时, lexsort的排序不是按照它原来的顺序来排序的.

$a = \left [ 1, 1, 3_{4}}, 4, 4, 4, 5_{1}}\right ]$

此时,对于数组a中的1, 我们根据其索引值到数组b中去找相同索引值所对应的数字,然后根据他们的大小来确定a中相同数字的排序顺序. 如a中1的索引值为0和2, 我们在b中找索引值为0和2的数字是9和2. 因为2小于9,所以a中索引为2对应的数字排在索引为0对应的数字之前. 如此得到如下二次排序的结果:

$a = \left [ 1_{2}, 1_{0}, 3_{4}, 4, 4, 4, 5_{1}}}}} \right ]$

依次,对于a中无法确定顺序的3个4. 注意他们在a中的索引值为:3,5,6. 我们在b中找相同索引所对应的数字.因为索引3对应的数字是4, 索引5对应的数字是2, 索引6对应的数字是1. 所以可以看到在b中, 索引6对应的数字<索引5对应的数字<索引3对应的数字. 相应的a中出现的顺序为:索引6对应的数字,索引5对应的数字,索引3对应的数字.于是,最终的排序如下:

$a = \left [ 1_{2}, 1_{0}, 3_{4}, 4_{6}, 4_{5}, 4_{3}, 5_{1}}}}}}}} \right ]$