Python numpy.argsort 排序索引

np.argsort可返回排序後的索引
一維列表:

>>> import numpy as np 
>>> a = [3,2,6,-1,5]
>>> np.argsort(a)
array([3, 1, 0, 4, 2], dtype=int64)

二維列表矩陣行排序

>>> b = [[2, -1, 5, 4],[3, 2, 0, 0],[-1, 6, -1, 2],[1, 1, 1, 1]]
>>> np.argsort(b)
array([[1, 0, 3, 2],
       [2, 3, 1, 0],
       [0, 2, 3, 1],
       [0, 1, 2, 3]], dtype=int64)

二維列排序>>> np.argsort(b,axis=0)
array([[2, 0, 2, 1],
       [3, 3, 1, 3],
       [0, 1, 3, 2],
       [1, 2, 0, 0]], dtype=int64)

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章