結果 : python if statement based on user input