python中找出矩阵每列的最大值的位置
时间: 2023-08-29 13:12:17 浏览: 112
求矩阵行最大值
可以使用NumPy库中的argmax函数来找出矩阵每列的最大值位置。示例如下:
```
import numpy as np
# 创建一个4行3列的随机矩阵
a = np.random.rand(4, 3)
# 找出每列的最大值位置
max_indices = np.argmax(a, axis=0)
# 输出每列的最大值位置
print(max_indices)
```
输出结果可能如下所示:
```
[3 0 1]
```
这表示第一列的最大值在第四行,第二列的最大值在第一行,第三列的最大值在第二行。
阅读全文