如何根据出现多次的月份来获取日期？

from datetime import datetime

def get_dates_by_month(months, start_date, end_date):
    date_format = "%Y-%m-%d"
    start = datetime.strptime(start_date, date_format)
    end = datetime.strptime(end_date, date_format)

    result = []
    current = start
    while current <= end:
        if current.month in months:
            result.append(current.strftime(date_format))
        current = current.replace(day=current.day + 1)

    return result

# 示例用法
months_to_find = [3, 6, 9]  # 要找的月份列表
start_date = "2022-01-01"  # 开始日期
end_date = "2022-12-31"  # 结束日期

dates = get_dates_by_month(months_to_find, start_date, end_date)
print(dates)