如何查找矩阵中元素最高的行

Question

我在python中使用高斯消除编写代码以创建下三角矩阵。到目前为止，我已经编写了一个部分来查找特定列中具有最高值的行，但是它始终返回错误的值，并且我看不到自己在做什么错。到目前为止，这是我的代码：

A = np.array([[1, 1, 1],
            [3, 1, 0],
            [2, 1, 0]], dtype = float)

n = len(A)

for i in range(0, n):
    max_element = abs(A[i][i])
    max_row = i
    for k in range(i+1, n):
        if abs(A[k][i]) > max_element:
            max_element = abs(A[k][I])
            max_row = k        

有人可以帮助我提出建议吗，请如何解决？

Answer 1

使用此

import numpy as np
A = np.array([[1, 1, 1],
            [3, 1, 0],
            [2, 1, 0]], dtype = float)

print(A[A.max(axis = 1).tolist().index(A.max()),:])

输出array([3., 1., 0.])

哪一行是3。

让我知道是否需要任何部分的说明。