Getting Numpy Array Size Python Interview Coding Challenges