Python Solve Matrix Equation