python寻找矩阵每列最大值的坐标
时间: 2023-07-09 14:51:46 浏览: 166
求矩阵行最大值
可以使用numpy库中的argmax函数来寻找矩阵每列的最大值的索引,然后使用np.unravel_index函数将索引转换成对应的行列坐标。以下是示例代码:
```python
import numpy as np
matrix = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
max_indices = np.argmax(matrix, axis=0) # 沿着列方向寻找最大值的索引
max_coords = np.unravel_index(max_indices, matrix.shape) # 将索引转换为坐标
print(max_coords)
```
在上述代码中,首先定义一个3x3的矩阵,然后使用argmax函数沿着列方向寻找最大值的索引,再使用np.unravel_index函数将索引转换为坐标,最后输出每列最大值的坐标。输出结果为:
```
(array([2, 2, 2], dtype=int64), array([2, 2, 2], dtype=int64))
```
其中,第一个数组为行坐标,第二个数组为列坐标,表示每列最大值的坐标。
阅读全文