User Input Array Size Python Programming