matrix multiplication in python user input using numpy