Numpy:使用numpy.sort()、numpy.argsort()获取排序后的值

获取矩阵排序后的值

 

print("---------------------np.sort()可得升序后的值-------------------------")
import numpy as np
a = np.array([[4, 3, 5], [1, 2, 1]])
print (a)
print("-----")

b = np.sort(a, axis=1)  # axis=1 表示按行排序,默认升序
print ("b=","\n",b)
print("-----")

a.sort(axis=1)
print ("a=",a)

print("------------np.argsort() 可获取从小到大的值的索引位置-----------------")
a = np.array([4, 3, 1, 2])
j = np.argsort(a)   # argsort 求的是从小到大的值的索引位置,即 j 返回的是索引
print ("j=",j)
print ("a[j]=",a[j])  # 得出从小到大排序后的值

 

 

 

结果图:

 

posted @ 2019-01-23 15:42  后来的后来123  阅读(8281)  评论(0编辑  收藏  举报