結果 : if statement based on input python