gpt4 book ai didi

python - 如何使用numpy降序排序?

转载 作者:太空狗 更新时间:2023-10-29 19:35:48 27 4
gpt4 key购买 nike

我有一个像这样的 numpy 数组:

A = array([[1, 3, 2, 7],
[2, 4, 1, 3],
[6, 1, 2, 3]])

我想按降序对这个矩阵的行进行排序,并像这样获取排序矩阵的参数:

As = array([[3, 1, 2, 0],
[1, 3, 0, 2],
[0, 3, 2, 1]])

我做了以下事情:

import numpy
A = numpy.array([[1, 3, 2, 7], [2, 4, 1, 3], [6, 1, 2, 3]])
As = numpy.argsort(A, axis=1)

但这给了我升序排序。此外,在我花了一些时间在互联网上寻找解决方案之后,我希望 numpy 中的 argsort 函数必须有一个参数可以颠倒排序顺序。但是,显然 没有这样的论点!为什么!?

有一个名为order 的参数。我通过猜测尝试了 numpy.argsort(..., order=reverse) 但它不起作用。

我在这里寻找以前问题的解决方案,我发现我可以做到:

import numpy
A = numpy.array([[1, 3, 2, 7], [2, 4, 1, 3], [6, 1, 2, 3]])
As = numpy.argsort(A, axis=1)
As = As[::-1]

出于某种原因,As = As[::-1] 没有给我所需的输出。

嗯,我想这一定很简单,但我遗漏了一些东西。

如何按降序对 numpy 数组进行排序?

最佳答案

只需将矩阵乘以 -1 即可反转顺序:

[In]: A = np.array([[1, 3, 2, 7],
[2, 4, 1, 3],
[6, 1, 2, 3]])
[In]: print( np.argsort(-A) )
[Out]: [[3 1 2 0]
[1 3 0 2]
[0 3 2 1]]

关于python - 如何使用numpy降序排序?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/36265930/

27 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com