通過Numpy的sort() 和 argsort()排序和恢復(fù)數(shù)組

在NumPy中萌朱,sort() 可以對數(shù)組進(jìn)行排序
argsort() 可以返回排序后的數(shù)組在原數(shù)組中的下標(biāo)

sort() 和 agrsort()

舉個例子:

原數(shù)組如下:

import numpy as np
a1 = np.array([10,6,8,2,5,4,9,1])
a1
## array([10,  6,  8,  2,  5,  4,  9,  1])

sort() 排序:

b1= np.sort(a1)
b1
##  array([ 1,  2,  4,  5,  6,  8,  9, 10])

argsort() 得到原來的索引位置

i1 = np.argsort(a1)
i1
## array([7, 3, 5, 4, 1, 2, 6, 0], dtype=int64)

那么如何通過這個索引位置恢復(fù)到原來的數(shù)組呢勺鸦?
則將上面的i1再進(jìn)行一次argsort()運(yùn)算覆积,得到索引认境,按照索引去排序過后的b1中找

b1[np.argsort(i1)]
##  array([10,  6,  8,  2,  5,  4,  9,  1])

題圖是在《Python One-Liners》這本書里的区拳,有興趣的朋友晨仑,可以看一下哦

參考資料:

  1. 書籍《Python One-Liners》
    2. https://stackoverflow.com/questions/2483696/undo-or-reverse-argsort-python
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末皮胡,一起剝皮案震驚了整個濱河市痴颊,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌屡贺,老刑警劉巖蠢棱,帶你破解...
    沈念sama閱讀 211,290評論 6 491
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場離奇詭異甩栈,居然都是意外死亡泻仙,警方通過查閱死者的電腦和手機(jī),發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 90,107評論 2 385
  • 文/潘曉璐 我一進(jìn)店門量没,熙熙樓的掌柜王于貴愁眉苦臉地迎上來玉转,“玉大人,你說我怎么就攤上這事殴蹄【孔ィ” “怎么了?”我有些...
    開封第一講書人閱讀 156,872評論 0 347
  • 文/不壞的土叔 我叫張陵饶套,是天一觀的道長漩蟆。 經(jīng)常有香客問我,道長妓蛮,這世上最難降的妖魔是什么怠李? 我笑而不...
    開封第一講書人閱讀 56,415評論 1 283
  • 正文 為了忘掉前任,我火速辦了婚禮蛤克,結(jié)果婚禮上捺癞,老公的妹妹穿的比我還像新娘。我一直安慰自己构挤,他們只是感情好髓介,可當(dāng)我...
    茶點(diǎn)故事閱讀 65,453評論 6 385
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著筋现,像睡著了一般唐础。 火紅的嫁衣襯著肌膚如雪箱歧。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 49,784評論 1 290
  • 那天一膨,我揣著相機(jī)與錄音呀邢,去河邊找鬼。 笑死豹绪,一個胖子當(dāng)著我的面吹牛价淌,可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播瞒津,決...
    沈念sama閱讀 38,927評論 3 406
  • 文/蒼蘭香墨 我猛地睜開眼蝉衣,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了巷蚪?” 一聲冷哼從身側(cè)響起病毡,我...
    開封第一講書人閱讀 37,691評論 0 266
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎钓辆,沒想到半個月后剪验,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 44,137評論 1 303
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡前联,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 36,472評論 2 326
  • 正文 我和宋清朗相戀三年功戚,在試婚紗的時候發(fā)現(xiàn)自己被綠了。 大學(xué)時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片似嗤。...
    茶點(diǎn)故事閱讀 38,622評論 1 340
  • 序言:一個原本活蹦亂跳的男人離奇死亡啸臀,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出烁落,到底是詐尸還是另有隱情乘粒,我是刑警寧澤,帶...
    沈念sama閱讀 34,289評論 4 329
  • 正文 年R本政府宣布伤塌,位于F島的核電站灯萍,受9級特大地震影響,放射性物質(zhì)發(fā)生泄漏每聪。R本人自食惡果不足惜旦棉,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 39,887評論 3 312
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望药薯。 院中可真熱鬧绑洛,春花似錦、人聲如沸童本。這莊子的主人今日做“春日...
    開封第一講書人閱讀 30,741評論 0 21
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽穷娱。三九已至绑蔫,卻和暖如春运沦,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背晾匠。 一陣腳步聲響...
    開封第一講書人閱讀 31,977評論 1 265
  • 我被黑心中介騙來泰國打工茶袒, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留,地道東北人凉馆。 一個月前我還...
    沈念sama閱讀 46,316評論 2 360
  • 正文 我出身青樓,卻偏偏與公主長得像亡资,于是被迫代替她去往敵國和親澜共。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 43,490評論 2 348