我有一个像这样的 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/