numpy -- argmax()、argmin()简介

NumPy的argmax()和argmin()函数用于返回数组中最大值和最小值的索引。

argmax()返回最大值索引，argmin()返回最小值索引，可以沿指定轴(axis=0/1)操作。

示例展示了在二维数组中查找全局极值索引，以及沿行(axis=1)和列(axis=0)方向查找极值索引的用法。这两个函数在处理多维数组时非常实用，能快速定位极值位置，是数据分析中的常用工具。

示例代码：

import numpy as np

x = np.array([[30, 40, 70], [80, 20, 10], [50, 90, 60]])
print('x数组：\n', x)
print('\n')

print('展开数组：')
print(x.flatten())
print('\n')

print('使用argmax()函数：\n', np.argmax(x))
print('\n')

print('沿轴0的最大索引：\n', np.argmax(x, axis=0))
print('\n')

print('沿轴1的最大索引：\n', np.argmax(x, axis=1))
print('\n')


print('使用argmin()函数：\n', np.argmin(x))
print('\n')

print('沿轴0的最小索引：\n', np.argmin(x, axis=0))
print('\n')

print('沿轴1的最小索引：\n', np.argmin(x, axis=1))
print('\n')

运行结果：