Python学习-numpy高级函数之sort-argmax-argmin函数

Numpy 高级函数一（sort、argmax、argmin）函数使用

首先导入numpy库

#导入numpy
import numpy as np

1.  numpy.sort()排序函数使用，次函数对数组中的元素进行排序

#sort 排序函数
a=np.array([[6,5],[8,9]])
print("a")
print(a)

输出结果：

a
[[6 5]
 [8 9]]

对数组a进行排序：

print("调用sort函数排序")
b=np.sort(a)
print(b)

输出结果：

调用sort函数排序
[[5 6]
 [8 9]]

2.numpy.argmax()函数使用，返回数组中最大值的索引，三种使用方式。

定义源数组：

#argmax函数，返回数组最大值的索引
a=np.array([[2,2,1],[4,6,5],[9,8,7]])
print("a")
print(a)

• 直接调用，返回所有元素最大值索引

print("调用argmax函数，直接调用返回最大值索引")
b=np.argmax(a)
print(b)

输出结果：

调用argmax函数，返回最大值索引
6

• 沿着0轴方向调用，输出每一行最大值索引

print("沿着0轴方向调用argmax")
print(np.argmax(a,axis=0))

输出结果：

沿着0轴方向调用argmax
[2 2 2]

• 沿着1轴方向调用，输出每一列最大值索引

print("沿着1轴方向调用argmax")
print(np.argmax(a,axis=1))

输出结果：

沿着1轴方向调用argmax
[0 1 0]

3.umpy.argmin()函数使用，返回数组中最小值索引，同argmax一样，也有直接调用，沿0轴调用和沿1轴调用三种调用方式；我这里就不一一演示了，直接把调用代码贴出。

首先定义一个用于操作的源数组：

#argmin函数，返回数组最小值索引
a=np.array([[1,2,4],[8,7,6],[0,3,9]])
print("a")
print(a)

下面是输出结果：

a
[[1 2 4]
 [8 7 6]
 [0 3 9]]

调用函数

print("直接调用argmix函数")
b=np.argmin(a)
print(b)

输出结果：
直接调用argmix函数
6


//沿0轴调用
print("沿着0轴方向调用argmin")
print(np.argmin(a,axis=0))

输出结果：
沿着0轴方向调用argmin
[2 0 0]


//沿1轴调用
print("沿着1轴方向调用argmin")
print(np.argmin(a,axis=1))


输出结果：
沿着1轴方向调用argmin
[0 2 0]