我正在用Python编写高斯-乔丹消去法的代码。我的指导如下：
def gauss_jordan(A):
for each row k do
i* <- argmax_{k<i<n} |A_{ik}|
if A_{i*k} = 0 then
Matrix is not invertible
end if
Swap rows k and i*
for each row j below k (i.e. j = k + 1,...,n) do
f = A_{jk}/A_{kk}
Aj = Aj - fA_{k}
end for
end for
for each row k = n,..., 1 (i.e. in reverse) do
A_{k} = A_{k}=A_{kk}
for each row j above k (i.e. j = k -1,..., 1) do
f = A_{jk}/A_{kk}
Aj = A_{j}-fA_{k}
end for
end for到目前为止我已经：
def gauss_jordan(A):
(h, w) = (len(A), len(A[0]))
for y in range(0,h):
for pivot in range(y, h):
if A[pivot][y].value % 2 != 0:
break
else:
return None这是正确的开始吗？我觉得很迷失在哪里去下一步。输入将是一个Numpy数组。任何想法都非常感谢！
1条答案
按热度按时间8ehkhllq1#
首先你要做的是创建增广矩阵。块的形式看起来像[A，恒等式（A.shape[0]）]，然后按照算法进行求解。您的最终答案将是矩阵的右半部分。我相信您的for循环是正确的，但检查不正确。您需要找到k列中的最大值。因此，当k为1时，您将遍历第一列并找到该列中绝对值的最大值并返回其索引。
注意，k是我的最外层循环，它遍历了增广矩阵A的所有行。希望这对你有帮助。