numpy学习笔记 np.argsort()

np.argsort()

>>> a=np.array([[1,2],[3,4]])
>>> a
array([[1, 2],
       [3, 4]])
>>> np.argsort(a,axis=1)
array([[0, 1],
       [0, 1]])
>>> np.argsort(a,axis=0)
array([[0, 0],
       [1, 1]])
>>> a=np.array([[1,2],[3,0]])
>>> np.argsort(a,axis=0)
array([[0, 1],
       [1, 0]])

np.argsort() 对数组进行排序，返回的数字是数组的元素在排序后所在的位置

np.newaxis

>>> a=np.arange(1,6)
>>> a
array([1, 2, 3, 4, 5])
>>> a.shape
(5,)
>>> b=a[:,np.newaxis]
>>> b
array([[1],
       [2],
       [3],
       [4],
       [5]])
>>> b.shape
(5, 1)

np.newaxis用来增加数组的维度，类似于pytorch的torch.unsqueeze()