相关疑难解决方法(0)

numpy 2d array max/argmax

我有一个numpy矩阵:

>>> A = np.matrix('1 2 3; 5 1 6; 9 4 2')
>>> A
matrix([[1, 2, 3],
        [5, 1, 6],
        [9, 4, 2]])
Run Code Online (Sandbox Code Playgroud)

我想获得每行中最大值的索引以及值本身.我可以使用A.argmax(axis = 1)得到最大值的索引,在这种情况下我会得到:

>>> indices = A.argmax(axis=1)
>>> indices
matrix([[2],
        [2],
        [0]])
Run Code Online (Sandbox Code Playgroud)

如何使用'indices'数组来获取矩阵中每行的最大值数组?有什么方法可以更有效地或在一次操作中做到这一点?是否有一个函数可以返回值及其行和列坐标?

python arrays numpy

6
推荐指数
1
解决办法
3709
查看次数

标签 统计

arrays ×1

numpy ×1

python ×1