多维数组求最大值索引
时间: 2023-07-11 15:21:26 浏览: 114
判断数组的最大值
3星 · 编辑精心推荐
可以使用以下代码来实现多维数组的最大值索引求解:
```python
import numpy as np
# 定义一个3x3的二维数组
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 获取最大值的索引
max_index = np.unravel_index(np.argmax(arr), arr.shape)
print("最大值的索引为:", max_index)
```
输出结果为:`最大值的索引为: (2, 2)`,表示最大值在第三行第三列的位置上。
如果是更高维度的数组,可以通过修改数组的 shape 参数来实现。
阅读全文