根据输入字母判断是星期几 Python

呜呜呜刚学习Python没多久，这个题还没掌握更简单的方法，导致我呕心沥血一个多小时才做出来，真的太不容易了

用户输入一个字母，如果这个字母在星期一到星期日的首字母中是唯一的，那就直接输出是星期几，如果不唯一就再让用户输入一个非首字母，然后再根据这两个字母进行判断输出星期几的英文单词，若输入2个字母不能确定是星期几输出提示“无法匹配”。 注：测试数据保证输入的字母都是表示星期的单词中存在的，第2个字母也必在可能的的答案的单词中存在。

代码实现

def get_weekday_first_letter(first_letter, second_letter=None):
    first_letter = first_letter.lower()
    if first_letter == 'm':
        return "Monday"
    elif first_letter == 't':
        if second_letter == 'e':
            return "Tuesday"
        elif second_letter == 'h' or second_letter == 'r' :
            return "Thursday"
        return "无法匹配"
    elif first_letter == 'w':
        return "Wednesday"
    elif first_letter == 'f':
        return "Friday"
    elif first_letter == 's':
        if second_letter == 't' or second_letter == 'r':
            return "Saturday"
        elif second_letter == 'n':
            return "Sunday"
        return "无法匹配"
    else:
        return "无法匹配"

def main():
    first_letter = input().lower()
    if first_letter == 't' or first_letter == 's':
        second_letter = input().lower()
        result = get_weekday_first_letter(first_letter, second_letter)
    else:
        result = get_weekday_first_letter(first_letter)
    print(result)

if __name__ == "__main__":
    main()

测试说明

平台会对你编写的代码进行测试：

测试输入： m 预期输出： Monday

测试输入： S t 预期输出： Saturday