Matrix multiplication exercises. 4 numpy arrays are multiplicated with each other in a loop.

matrix/exercises.py

@@ -0,0 +1,26 @@
+import numpy as np
+
+a = np.array([[1], [2]])
+b = np.array([[0, 1, -1]])
+A = np.array([[0, 1], [-1, 1]])
+B = np.array([[2, 0, 1], [1, -1, 0]])
+
+calculations = [
+    lambda: np.dot(A, a),
+    lambda: np.dot(a, B),
+    lambda: np.dot(a.T, B),
+    lambda: np.dot(B, b.T),
+    lambda: np.dot(A, B),
+    lambda: np.dot(np.dot(a.T, B), A),
+    lambda: np.dot(np.dot(a.T, B).T, A),
+    lambda: np.dot(B, A),
+    lambda: np.dot(B.T, A),
+]
+
+for calc in calculations:
+    try:
+        print(calc())
+        print("\n")
+    except Exception as e:
+        print(e)
+        print("\n")