User Input Array Size Python Code Examples