from datetime import datetime, timedelta
def create_month_list(start_month, num_months):
"""创建月份列表"""
month_list = []
for i in range(num_months):
if i == 0:
# 如果是第一个月,则添加日期格式的1月
month_list.append(datetime.strftime(start_month, '%Y-%m-%d'))
else:
# 对于其余的月份,则添加字符串格式的月份
month = start_month + timedelta(days=31*i)
month_list.append(datetime.strftime(month, '%Y-%B'))
return month_list
month_list = create_month_list(datetime(2021, 1, 1), 12)
print(month_list)
输出:
['2021-01-01', '2021-February', '2021-March', '2021-April', '2021-May', '2021-June', '2021-July', '2021-August', '2021-September', '2021-October', '2021-November', '2021-December']
该函数通过传入起始月份和需要创建的月份数量来创建包含月份的列表。对于第一个月,即1月,使用日期格式。其余月份则使用字符串格式。