ส่งงาน week05

week05/ge.py

@@ -2,7 +2,38 @@
 #นายกิตติศักดิ์ มณีวงษ์
 
 def ge(A,b):
-    pass
+    import numpy as np
+    A=np.array(A)
+    b=np.array(b)
+    n = len(b)
+    for k in range(0, n - 1):
+        for i in range(k + 1, n):
+            lam = A[i, k] / A[k, k]
+            A[i, k] = 0
+            A[i, k + 1:n] = A[i, k + 1:n] - lam * A[k, k + 1:n]
+            b[i] = b[i] - lam * b[k]
+    # back
+    x = b.copy()
+    print(x)
+    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]
+    print(x)
+
 
 def solve(A,b):
-    pass
\ No newline at end of file
+    import numpy.linalg
+    result = numpy.linalg.solve(A,b);
+    for i,x in zip(range(1,len(result)+1),result) :
+        print("x_{} = {}".format(i,x))
+
+#5x3x1
+m,n,_=map(int,input().split('x'))
+print(m,n)
+A = []
+for row in range(m):
+    #2,3,1,4,7
+    A.append(list(map(int,input().split(','))))
+print(n,_)
+b = list(map(int,input().split(',')))
+solve(A,b)
+ge(A,b)