1. 程式人生 > >Numpy:使用numpy.sort()、numpy.argsort()獲取排序後的值

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

ron -- array ima args \n nbsp 索引 img

獲取矩陣排序後的值

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]) # 得出從小到大排序後的值

結果圖:

技術分享圖片

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