numpy().argsort()函数

原创于 2021-12-15 20:49:02 发布 · 1.3k 阅读

CC 4.0 BY-SA版权

文章标签：

7 篇文章

订阅专栏

本文详细介绍了numpy的sort和argsort函数在处理一维和二维数组时的应用，包括升序和降序排列，以及如何根据轴(axis)进行行和列排序。通过实例演示了这些功能在实际编程中的使用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import numpy as np

a=np.array([5,1,3])

b=np.argsort(a)
print(b)

输出结果：

[1 2 0]

argsort(a,axis=0):按列排序

argsort(a,axis=1):按行排序

import numpy as np


a=np.array([[5,3],[2,1]])

b=np.argsort(a,axis=0)
print(b)

c=np.argsort(a,axis=1)
print(c);

输出结果

[[1 1]
 [0 0]]
[[1 0]
 [1 0]]

argsort(a)：按升序排列

argsort(-a):按降序排列

import numpy as np


a=np.array([5,3,-2,8])

print(np.argsort(a))

print(np.argsort(-a))

输出结果:

[2 1 0 3]
[3 0 1 2]