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))

运行结果为:

猜你喜欢

转载自blog.csdn.net/Sophia_11/article/details/84978772