Put Numpy Array Size Python Code Writer