import numpy as np a= np.array([3, 1, 2]) b=np.argsort(a) print(a,b,a[b]) ''' numpy.argsort 将数组按照某个维度升序排列(从小到大排列) 排列之后的索引值是在原数组中的索引值 a=np.array([a[0],a[1],a[2]]) 排序之后的数组应该是: np.sort(a)=np.array([a[1],a[2],a[0]) 故而输出的索引值是 np.argsort(a)=[1,2,0] a[np.argsort(a)]=np.array(a[1],a[2],a[0])=[1,2,3] ''' a=np.array([3,1,2,4]) b=np.argsort(a)# [1,2,0,3] print(a,b,a[b])# [1,2,3,4]
np.argsort(a)
最新推荐文章于 2024-06-06 08:10:55 发布