Python使用NumPy对数组排序并获取对应索引值的方法

咖啡半糖
咖啡半糖
发布时间:2026-10-13 | 2人浏览过
原创

本文介绍在Python中利用NumPy库对数组进行排序并获取对应索引值的方法。通过np.sort()实现数值排序,利用np.argsort()分别获取正序和逆序的索引序列。内容涵盖一维与二维数组的处理,包括按行或按列排序的具体操作,帮助开发者快速掌握数组排序及索引提取技巧。

Python数组排序与索引获取基础

以下直接展示相关代码实现。


# -*- coding: cp936 -*-
import numpy as np

#一维数组排序
arr = [1, 3, 5, 2, 4, 6]
arr = np.array(arr)
print arr
print np.sort(arr)#或print np.sort(arr,axis=None)

print (np.argsort(arr)) # 正序输出索引,从小到大
print (np.argsort(-arr)) # 逆序输出索引,从大到小

运行后的具体输出如下:


[1 3 5 2 4 6]
[1 2 3 4 5 6]
[0 3 1 4 2 5]
[5 2 4 1 3 0]

#二维数组排序
list1 = [[4,3,2],[2,1,4]]
array=np.array(list1) 
print array
array.sort(axis=1) #axis=1按行排序,axis=0按列排序
print array

另一组测试的输出结果如下:


[[4 3 2]
[2 1 4]]

[[2 3 4]
[1 2 4]]

数组排序并保留索引的补充实现

补充内容:Python数组排序并保留索引的处理方式

具体代码示例如下:


import numpy as np

arr = [1, 3, 5, 2, 4, 6]
arr = np.array(arr)
print (np.argsort(arr)) # 正序输出
print (np.argsort(-arr)) # 逆序输出

以上就是Python中对数组进行排序并输出对应索引值的完整方法。希望上述内容能为读者提供有价值的参考,欢迎持续关注与支持。

相关标签:

本站声明

本文内容仅供参考,不代表本站观点。转载或引用请注明出处。如发现内容问题,请通过联系我们页面反馈。

Python 相关产品和应用

全部下载

专题列表