Getting Array input from user using for Loop in Python