计算数组中元素的最大值及其所在的行列下标值
时间: 2023-04-29 08:04:25 浏览: 118
可以使用numpy库中的numpy.argmax()函数来计算数组中元素的最大值所在的下标,其中返回的是一维下标。如果要得到二维数组中的行列下标,可以使用numpy.unravel_index()函数将一维下标转换为二维行列下标。
例如:
```
import numpy as np
a = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
max_index = np.argmax(a)
row, col = np.unravel_index(max_index, a.shape)
print("最大值为:", a[row, col])
print("所在行:", row)
print("所在列:", col)
```
输出:
```
最大值为: 9
所在行: 2
所在列: 2
```
相关问题
计算数组中元素最大值及其所在的行列下标值
可以使用以下方法计算数组中元素最大值及其所在的行列下标值:
1. 定义一个变量来存储最大值,初始值为数组中的第一个元素。
2. 遍历数组,比较每个元素与最大值的大小,如果当前元素大于最大值,则将最大值更新为当前元素。
3. 在遍历过程中,记录最大值所在的行列下标值,可以使用两个变量来分别存储。
4. 遍历完成后,返回最大值及其所在的行列下标值。
以下是示例代码:
```python
def find_max(arr):
max_val = arr[][]
max_row =
max_col =
for i in range(len(arr)):
for j in range(len(arr[i])):
if arr[i][j] > max_val:
max_val = arr[i][j]
max_row = i
max_col = j
return max_val, max_row, max_col
```
其中,arr为输入的二维数组。函数返回一个元组,包含最大值及其所在的行列下标值。
计算数组中元素的最大值及其所在行列下标值
可以使用两个变量来记录最大值和其所在的行列下标值,遍历数组,如果当前元素大于最大值,则更新最大值和行列下标值。最后返回最大值和行列下标值即可。以下是示例代码:
```python
def find_max(arr):
max_val = arr[0][0]
max_row = 0
max_col = 0
for i in range(len(arr)):
for j in range(len(arr[0])):
if arr[i][j] > max_val:
max_val = arr[i][j]
max_row = i
max_col = j
return max_val, max_row, max_col
```
使用示例:
```python
arr = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
max_val, max_row, max_col = find_max(arr)
print("最大值为:", max_val)
print("所在行列下标值为:", max_row, max_col)
```
输出结果:
```
最大值为: 9
所在行列下标值为: 2 2
```
阅读全文