numpy求二位数组行的最大值索引
时间: 2024-05-02 07:22:12 浏览: 198
可以使用numpy的argmax函数来求二维数组行的最大值索引。具体操作如下:
```python
import numpy as np
# 创建一个3行4列的二维数组
arr = np.array([[1, 2, 3, 4],
[5, 6, 7, 8],
[9, 10, 11, 12]])
# 求每行的最大值索引
max_index = np.argmax(arr, axis=1)
print(max_index) # 输出 [3 3 3]
```
其中,`axis=1`表示沿着行的方向求最大值索引。结果显示每一行的最大值索引分别是3,3,3,即第4列的索引。
阅读全文