Python使用NumPy对数组排序并获取对应索引值的方法
本文介绍在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中对数组进行排序并输出对应索引值的完整方法。希望上述内容能为读者提供有价值的参考,欢迎持续关注与支持。
本站声明
本文内容仅供参考,不代表本站观点。转载或引用请注明出处。如发现内容问题,请通过联系我们页面反馈。













