numpy二维数组最大值_求二维数组的最大值及其行下标

激活谷笔记 • 2026-05-18 09:10 • 阅读 2

在Python中，要找到二维数组的最大值，你可以使用多种方法。以下是几种常见的方法：

1. 使用嵌套循环遍历数组：

python

arr = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]

max_val = float（'-inf'）初始化最大值为负无穷

for i in range（len（arr））:

for j in range（len（arr[i]））:

if arr[i][j] > max_val:

max_val = arr[i][j]

print（max_val）输出最大值

2. 使用`max`函数，但需要注意`max`函数默认是按照素的第一个值排序的，所以需要指定`axis`参数：

python

a = [[1, 2], [2, 5], [3, 4]]

max_val = max（a, axis=1）按行求最大值

print（max_val）输出最大值

3. 使用`numpy`库的`amax`函数：

python

import numpy as np

x = np.array（[[1, 2, 3], [4, 5, 6], [7, 8, 9]]）

max_all = np.amax（x）求整个数组的最大值

max_dimension1 = np.amax（x, axis=0）求每列的最大值

max_dimension2 = np.amax（x, axis=1）求每行的最大值

print（'a:\n', a）

print（'max_all:', max_all）

print（'max_dimension1:', max_dimension1）

print（'max_dimension2:', max_dimension2）

4. 使用`map`函数结合`max`函数：

python

a = [[1, 2], [2, 5], [3, 4]]

max_val = max（map（max, a））

print（max_val）输出最大值

5. 使用`numpy`库的`argmax`函数获取最大值的索引，然后通过索引获取最大值：

python

import numpy as np

x = np.array（[[1, 2, 3], [4, 5, 6], [7, 8, 9]]）

index_max = np.argmax（x, axis=1）获取每行的最大值索引

max_val = x[range（x.shape）, index_max] 通过索引获取最大值

print（max_val）输出最大值

以上方法都可以用来找到二维数组的最大值。选择哪种方法取决于你的具体需求以及你对Python和`numpy`库的熟悉程度