我们可以编写一个函数来解决这个问题,该函数接受一个日期字符串作为参数并返回相应的季节字符串。以下是示例代码:
def get_season(date_str):
# 将日期字符串转换为日期对象
try:
date = datetime.datetime.strptime(date_str, '%Y-%m-%d')
except ValueError:
return 'Invalid date'
# 根据月份判断季节
month = date.month
if month in [3, 4, 5]:
return 'Spring'
elif month in [6, 7, 8]:
return 'Summer'
elif month in [9, 10, 11]:
return 'Autumn'
else:
return 'Winter'
调用这个函数的方式很简单,只需要提供一个符合格式的日期字符串作为参数即可。例如:
>>> get_season('2022-03-20')
'Spring'
>>> get_season('2022-06-21')
'Summer'
>>> get_season('2022-13-01')
'Invalid date'