User Input Array Size Python Code Practices