挑选出每一行最大的数的列索引的Python代码
时间: 2024-03-20 18:39:09 浏览: 69
Python 实现Numpy中找出array中最大值所对应的行和列
可以使用以下代码来挑选出每一行最大的数的列索引:
```python
import numpy as np
# 创建一个矩阵
matrix = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 使用argmax函数来获取每一行最大的数的列索引
max_index = matrix.argmax(axis=1)
print(max_index)
```
输出结果为:
```
[2 2 2]
```
其中,`axis=1` 表示沿着每一行进行操作。`max_index` 中的每个元素表示对应行中最大值的列索引。
阅读全文