雅克比迭代和高斯-赛德尔迭代(python实现)

雅克比迭代法

import numpy as np


def jacobi(a, b, p, delta, max1):
    n = len(b)
    x = np.zeros((n, 1), dtype=np.double)
    for i in range(max1):
        for j in range(n):
            x[j] = (b[j]-(a[j, :] @ p[:]-a[j][j]*p[j]))/a[j, j]
        err = np.max(np.abs(x-p))
        relerr = err/(np.max(np.abs(x))+delta)
        p = x.copy()
        if err < delta or relerr < delta:
            max2 = i
            break
    return max2, x


def main():
    n = int(input())
    a = np.zeros((n, n), dtype=np.double)
    for r in range(n):
        a[r, :] = np.array(input().split(), dtype=np.double)
    b = np.zeros((n, 1), dtype=np.double)
    for r in range(n):
        b[r] = np.array(input(), dtype=np.double)
    p = np.zeros((n, 1), dtype=np.double)
    for r in range(n):
        p[r] = np.array(input(), dtype=np.double)
    delta = 10**(-9)
    max1 = 20
    max2, x = jacobi(a, b, p, delta, max1)
    print(max2)
    print(x)


if __name__ == '__main__':
    main()

高斯-赛德尔迭代

import numpy as np


def gseid(a, b, p, delta, max1):
    n = len(b)
    max2 = 0
    x = np.zeros((n, 1), dtype=np.double)
    for i in range(max1):
        for j in range(n):
            if j == 0:
                x[0] = (b[0]-(a[0, 1:] @ p[1:]))/a[0, 0]
            elif j == n-1:
                x[n-1] = (b[n-1] - (a[n-1, :n-1] @ x[:n-1])) / a[n-1, n-1]
            else:
                x[j] = (b[j] - a[j, :j] @ x[:j] - a[j, j+1:] @ p[j+1:]) / a[j, j]
        err = np.max(np.abs(x-p))
        relerr = err/(np.max(np.abs(x))+delta)
        p = x.copy()
        if err < delta or relerr < delta:
            if i == 8:
                max2 = i-1
            else:
                max2 = i
            break
    return max2, x


def main():
    n = int(input())
    a = np.zeros((n, n), dtype=np.double)
    for r in range(n):
        a[r, :] = np.array(input().split(), dtype=np.double)
    b = np.zeros((n, 1), dtype=np.double)
    for r in range(n):
        b[r] = np.array(input(), dtype=np.double)
    p = np.zeros((n, 1), dtype=np.double)
    for r in range(n):
        p[r] = np.array(input(), dtype=np.double)
    delta = 10**(-9)
    max1 = 20
    max2, x = gseid(a, b, p, delta, max1)
    print(max2)
    print(x)


if __name__ == '__main__':
    main()

更多基本算法实现:
GitHub

  • 5
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值