How To Take User Input Of Array In Python