Numpy Matrix Into Array Interview