查找从特定日期开始的月份

import datetime

def find_months_from_date(start_date, num_months):
    # 解析特定日期
    year, month, _ = start_date.split('-')
    year = int(year)
    month = int(month)

    # 获取当前日期
    current_date = datetime.date(year, month, 1)

    # 遍历月份
    for _ in range(num_months):
        # 打印当前月份
        print(current_date.strftime('%Y-%m'))

        # 计算下一个月份
        if month == 12:
            year += 1
            month = 1
        else:
            month += 1

        # 更新当前日期
        current_date = datetime.date(year, month, 1)

# 示例调用
find_months_from_date('2022-03-15', 6)

2022-03
2022-04
2022-05
2022-06
2022-07
2022-08