공학수학 1: Misc change

simnalamburt · May 4, 2021 · b57f81d · b57f81d
1 parent a18fd5d
commit b57f81d
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 11 deletions.
diff --git a/공학수학 1/gauss-seidel.py b/공학수학 1/gauss-seidel.py
@@ -13,14 +13,15 @@
 ]
 
 B = [
-    50,
-    50,
-    25,
-    25,
+    50.,
+    50.,
+    25.,
+    25.,
 ]
 
 # Initial values
-u11, u21, u12, u22 = 0, 0, 0, 0
+u11, u21, u12, u22 = 0., 0., 0., 0.
+print(f'U = [{u11} {u21} {u12} {u22}]')
 
 for _ in range(20):
     v11 = A[0][0]*u11 + A[0][1]*u21 + A[0][2]*u12 + A[0][3]*u22 + B[0]

diff --git a/공학수학 1/jacobi.py b/공학수학 1/jacobi.py
@@ -15,15 +15,17 @@
 ])
 
 B = np.array([
-    50,
-    50,
-    25,
-    25,
+    50.,
+    50.,
+    25.,
+    25.,
 ])
 
 # Initial values
-U = np.array([0, 0, 0, 0])
+U = np.array([0., 0., 0., 0.])
+print(f'U = {U}')
 
 for _ in range(20):
-    U = A@U + B
+    V = A@U + B
+    U = A@V + B
     print(f'U = {U}')