淺述python中argsort()函數(shù)的用法

? ? 由于想使用python用訓(xùn)練好的caffemodel來對(duì)很多圖片進(jìn)行批處理分類，學(xué)習(xí)過程中，碰到了argsort函數(shù)，因此去查了相關(guān)文獻(xiàn)，也自己在python環(huán)境下進(jìn)行了測(cè)試，大概了解了其相關(guān)的用處，為了怕自己后面又忘了，就寫下來權(quán)當(dāng)加深理解了。(ps:我也是python小白，理解可能比較淺顯)

1.先定義一個(gè)array數(shù)據(jù)

1 import numpy as np 2 x=np.array([1,4,3,-1,6,9])

2.現(xiàn)在我們可以看看argsort()函數(shù)的具體功能是什么：

x.argsort()

輸出定義為y=array([3,0,2,1,4,5])。

我們發(fā)現(xiàn)argsort()函數(shù)是將x中的元素從小到大排列，提取其對(duì)應(yīng)的index(索引)，然后輸出到y(tǒng)。例如：x[3]=-1最小，所以y[0]=3,x[5]=9最大，所以y[5]=5。

上面這個(gè)不難理解，不熟悉的可以去python環(huán)境下自己嘗試。

3.由于在程序中遇到了類似于np.argsort()[num]的形式，一直看不明白，就自己去python環(huán)境自己試了下：

ps：這里的num的絕對(duì)值小于等于x中元素的個(gè)數(shù)

當(dāng)num>=0時(shí)，np.argsort()[num]就可以理解為y[num];

當(dāng)num<0時(shí)，np.argsort()[num]就是把數(shù)組y的元素反向輸出，例如np.argsort()[-1]即輸出x中最大值對(duì)應(yīng)的index，np.argsort()[-2]即輸出x中第二大值對(duì)應(yīng)的index，依此類推。。

直觀的實(shí)驗(yàn)才能看到效果，下面是我拿上面例子做的驗(yàn)證：

這是當(dāng)num為負(fù)值時(shí)的輸出。

這個(gè)是num>=0時(shí)的輸出。

總結(jié)：根據(jù)上面的闡述，應(yīng)該對(duì)argsort()的用法有了一個(gè)簡(jiǎn)單的了解。由于我沒有去看官方解釋，所以理解可能欠缺，希望諸位指正。

總結(jié)

以上是生活随笔為你收集整理的浅述numpy中argsort()函数的用法的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。