Python函数表达

日历结果不是输出某个月的日历是输出指定年月最后一天为周几
用函数表达

def func(year):
    import datetime
    _ = ["一", "二", "三", '四', '五', '六', '日']
    strtime = year+"/12/31"
    week = datetime.datetime.strptime(strtime, '%Y/%m/%d').weekday()

    return "周" + _[week]


year = input("请输入年份")
print(func(year))