Create example.py

CompPhysics · Feb 17, 2024 · 3854f12 · 3854f12
1 parent dace3c4
commit 3854f12
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/doc/src/week6/programs/example.py b/doc/src/week6/programs/example.py
@@ -0,0 +1,12 @@
+import numpy as np
+
+# Solution by brute force inversion
+A = np.array([[2,1],[1,20]])
+b = np.array([5,3])
+invA = np.linalg.inv(A)
+x = invA @ b
+print(x)
+print(np.linalg.eig(A))
+#  Solution with iterative methods
+
+# Solution with gradient descent and fixed learning rate