ES.1803 pset 6 problem 1 checker

es.1803-pset6-1.py

import numpy as np
import sympy as sp

a = [[0, 0, 1, 0], [0, 0, 0, 1], [-5, 1, 0, 0], [1, -4, 0, 0]]
b = [[1, 2, 3], [0, 2, 1], [4, 2, 0]]
A = np.array(a)
B = np.array(b)
A_sp = sp.Matrix(a)
B_sp = sp.Matrix(b)
b = np.array([2, 0, 5, 0])

# a
Ab = A @ b
print("a.")
print(Ab)
print("===")

# b
# (dimensions error)
# BA = B @ A
# print("b.")
# print(BA)
# print("===")

# c
det_A = np.linalg.det(A)
print("c.")
print(det_A)
print("===")

# d
inv_A = np.linalg.inv(A)
print("d.")
print(inv_A)
print("===")

# e
x = inv_A @ b
print("e.")
print(x)
print("===")

# f
rref_A, _ = A_sp.rref()
rref_B, _ = B_sp.rref()
print("f.")
print(rref_A)
print(rref_B)
print("===")

# g
AT = A.T
BT = B.T
print("g.")
print(AT)
print(BT)
print("===")