Numpy中argsort()函數(shù)的用法

argsort()函數(shù)的作用是將數(shù)組按照從小到大的順序排序，并按照對(duì)應(yīng)的索引值輸出。

argsort()函數(shù)中，當(dāng)axis=0時(shí)，按列排列；當(dāng)axis=1時(shí)，按行排列。如果省略默認(rèn)按行排列。

下邊通過(guò)例子來(lái)說(shuō)明其用法

#!/usr/bin/env python
# -*- coding:utf-8 -*-
from numpy import *
a = [3, 5, 1]
b = argsort(a)
print("a排序?yàn)椋?, end='')
print(b)
c = [[8, 3, 7], [1, 5, 9]]
d = argsort(c)  # 默認(rèn)按行排序
print("默認(rèn)按行排序c為：")
print(d)
print("按行排序c為：")
print(argsort(c, axis=1))
print("按行列排序c為：")
print(argsort(c, axis=0))

運(yùn)行結(jié)果為：

Numpy.argsort()函數(shù)使用（以按矩陣指定列對(duì)各行排序?yàn)槔?/h2>

該方法返回的是矩陣a要素排序后的索引數(shù)據(jù)，干說(shuō)無(wú)用，以二維數(shù)組為例，按照指定列，如第0列，進(jìn)行排序。

Python代碼： 

首先看代碼，再逐行解釋。

import numpy as n
a = np.array([[0, 11, 12], [2, 2, 3], [7, 8, 9], [1, 2, 3]])
ind = np.argsort(a, axis=0)
print(a)
print(ind)
print(a[ind[:, 0]])
# Output:
# [[ 0 11 12]
#  [ 2  2  3]
#  [ 7  8  9]
#  [ 1  2  3]]
# [[0 1 1]
#  [3 3 3]
#  [1 2 2]
#  [2 0 0]]
# [[ 0 11 12]
#  [ 1  2  3]
#  [ 2  2  3]
#  [ 7  8  9]]

待排序數(shù)組"a"：

[[ 0 11 12]
[ 2 2  3]
[ 7 8   9]
[ 1 2  3]] 

使用Numpy.argsort對(duì)a的各行進(jìn)行比較，首先會(huì)比較a每行的第一列的數(shù)值，按第一列數(shù)值從小到大排序，再取第二列的值比較，如此往復(fù)，最后返回各個(gè)元素的在a中的索引值矩陣，將a各元素的索引值放到排序后的位置，你將得到這樣一個(gè)索引矩陣"ind"：

索引矩陣"ind"：

[[[0 1 1]
[[3 3 3]
[[1 2 2]
[ [2 0 0]] 

第0列[0 3 1 2]分別表示a[0][0], a[0][3], a[0][1], a[0][2]在a第0列的索引值，同理ind的第1列表示在a第1列的索引值，由于a[0][0]最小所以它還是在第一個(gè)位置，a[0][2]最大，所以被放置到了最后一個(gè)位置，所以a[0][2]的索引2就在ind矩陣第0列中的最后一個(gè)位置。

最后，我們可以取ind的第0列，將矩陣a的各行按照這列的索引排序，即第0行不變，第1行放原先第3行[1 2 3]，第2行放原先第1行[2 2 3]，第3行放原先第2行[7 8 9]，排序后的數(shù)組如下：

排序后的數(shù)組"a"：

[[[ 0 11 12]
[ [ 1 2  3]
[ [ 2 2  3]
[ [ 7 8  9]]]

總結(jié)

以上為個(gè)人經(jīng)驗(yàn)，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。

最新評(píng)論