import numpy as np
初始化係數矩陣A和常數項向量b
n = 1000
A = np.zeros((n, n))
b = np.arange(1, n+1)
填充係數矩陣A
for i in range(n):
A[i, i] = 4 # 對角線元素為4
if i < n-1:
A[i, i+1] = 1 # 每一行的下一個元素為1
if i > 0:
A[i, i-1] = 1 # 每一行的上一個元素為1(除了第一行)
求解線性方程組 Ax = b
x = np.linalg.solve(A, b)
列印解
print("解向量 x:")
for i in range(n):
print(f"x{i+1} = {x[i]:.4f}")
print("學號:3025")