1. 程式人生 > >Numpy中argsort()函式的用法

Numpy中argsort()函式的用法

argsort()函式的作用是將陣列按照從小到大的順序排序,並按照對應的索引值輸出

argsort()函式中,當axis=0時,按列排列;當axis=1時,按行排列。如果省略預設按行排列


下邊通過例子來說明其用法:

#!/usr/bin/env python
# -*- coding:utf-8 -*-
from numpy import *

a = [3, 5, 1]
b = argsort(a)
print("a排序為:", end='')
print(b)

c = [[8, 3, 7], [1, 5, 9]]
d = argsort(c)  # 預設按行排序
print("預設按行排序c為:")
print(d)

print("按行排序c為:")
print(argsort(c, axis=1))

print("按行列排序c為:")
print(argsort(c, axis=0))

執行結果為: