ส่งงาน week05

week05/ge.py

 # 59110440039
 # นายกิตติพงษ์ มณีวงษ์
-def ge(A,b):
-    pass
+iimport numpy as np
+def ge(A,b) :
+  n=len(b)
+  for k in range(0,n-1):
+    for i in range(k+1,n):
+        if A[i,k] != 0.0:
+            lam = A[i,k]/A[k,k]
+            A[i,k] = 0
+            A[i,k+1:n] -= lam*A[k, k+1:n]
+            b[i] -= lam*b[k]
+  x=b.copy()
+  for k in range(n-1, -1, -1):
+    x[k] = (b[k] - np.dot(A[k,k+1:n], x[k+1:n]))/A[k,k]
+
+  return (x)
+
+A = np.array([
+        [4, -2, 1],
+        [-2, 4, -2],
+        [1, -2, 4]
+    ], float)
+b = np.array([11,-16,17], float)
+print(ge(A,b))
 
 def solve(A,b):
     import numpy.linalg