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]) # 得出从小到大排序后的值
结果图: