Getting Numpy Array Size Python Code For Calculator