User Input Array Size Python Code For Rock