【Numpy】数组极值

1.一维数组极值以及对应的下标索引

• np.max 和 np.min用于查找最大值和最小值
• np.argmax 和 np.argmax用于查找极值元素对应的下标索引

#查找数组的最大值、最小值以及对应的下标
import numpy as np
a = np.array([3, 1, 7, 4, 2, 5, 8])
print(np.max(a))	# 8
print(np.min(a))	# 1
print(np.argmax(a))	# 6
print(np.argmin(a))	# 1

2.二维数组极值

• 设置axis=0，表示沿着竖直方向(行索引增长的方向)，为每一列分别查找极值(一维数组，长度为原二维数组列数)
• 设置axis=1，表示沿着水平方向(列索引增长的方向)，为每一行分别查找极值(一维数组，长度为原二维数组行数)
• 不设置axis时，将查找整个数组中的极值元素(单个值，标量)

#查找数组的最大值，每行的最大值，每列的最大值
a = np.array([[1,2,3],[4,5,6]])
print(np.max(a))			# 6
print(np.max(a,axis = 1))	# [3 6]    每行的最大值
print(np.max(a,axis = 0))	# [4 5 6]  每列的最大值

3.二维数组极值对应的元素索引

• argmax和argmin返回极值对应的索引
• axis=1时，分别返回每个行的极值在该行中的索引号
• axis=0时，分别返回每个列的极值在该列中的索引号
• 如果没有指定axis，则返回极值元素在整个二维数组中的索引号(将二维数组展成一维数组)

#查找数组的最大值，每行的最大值，每列的最大值对应的下标
a = np.array([[1,2,3],[4,5,6]])
print(np.argmax(a))				# 5
print(np.argmax(a,axis = 1))	# [2 2]
print(np.argmax(a,axis = 0))	# [1 1 1]