np.argsort()怎么使用

發(fā)布時間：2021-12-27 14:29:46 來源：億速云閱讀：234 作者：iii 欄目：大數(shù)據(jù)

本篇內(nèi)容主要講解“np.argsort()怎么使用”，感興趣的朋友不妨來看看。本文介紹的方法操作簡單快捷，實用性強。下面就讓小編來帶大家學(xué)習(xí)“np.argsort()怎么使用”吧!

numpy.argsort(a, axis=-1, kind=’quicksort’, order=None)

功能: 將矩陣a按照axis排序，并返回排序后的下標
參數(shù): a:輸入矩陣， axis:需要排序的維度
返回值: 輸出排序后的下標

(一維數(shù)組)

import numpy as np

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

x.argsort()

# array([3, 0, 1, 2, 4, 5], dtype=int64)

可以發(fā)現(xiàn)，argsort()是將X中的元素從小到大排序后，提取對應(yīng)的索引index，然后輸出到y(tǒng)
如x[3]=-1最小，x[5]=9最大

所以取數(shù)組x的最小值可以寫成:

x[x.argsort()[0]]

或者用argmin()函數(shù)

x[x.argmax()]

數(shù)組x的最大值，寫成：

x[x.argsort()[-1]]  # -1代表從后往前反向的索引

或者用argmax()函數(shù),不再詳述

x[x.argmax()]

輸出排序后的數(shù)組

x[x.argsort()]
# 或
x[np.argsort(x)]

(二維數(shù)組)

x = np.array([[1,5,4],[-1,6,9]])
# [[ 1  5  4]
# [-1  6  9]]

沿著行向下(每列)的元素進行排序

np.argsort(x,axis=0)
# array([[1, 0, 0],
#        [0, 1, 1]], dtype=int64)

沿著列向右(每行)的元素進行排序

np.argsort(x,axis=1)
# array([[0, 2, 1],
#        [0, 1, 2]], dtype=int64)

到此，相信大家對“np.argsort()怎么使用”有了更深的了解，不妨來實際操作一番吧！這里是億速云網(wǎng)站，更多相關(guān)內(nèi)容可以進入相關(guān)頻道進行查詢，關(guān)注我們，繼續(xù)學(xué)習(xí)！

向AI問一下細節(jié)

猜你喜歡