Python Numpy Create Matrix Word Problems