how to do matrix addition and subtraction using python