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))
运行结果为: