Valueerror: could not broadcast input array from shape (2,2) into shape (2)

Question

I\'m writing python code to do finite element analysis and make a stiffness matrix. It threw this error and I just can\'t figure out why.

def solve_displacement(a