Getting Numpy Array Size Python Interview Bit