[PYTHON]7-3 程序员买苹果

老婆给程序员老公发了一条信息:下班买6个苹果回来,如果看到西瓜,就买1个，~ 结果老公回去的时候,手上仅拿着一个苹果，问不是让你买6个吗？答曰，因为看到西瓜了，就只买1个。本题目要求读入一个小写字母，然后根据字母是'y'则输出1,如果是'n'则输出6,若是其他字符，输出error

x = input()
if x == 'y':
    print(1)
elif x == 'n':
    print(6)
else:
    print('error')