如何根据日期按周计数

import datetime

def count_weeks(start_date, end_date):
    start_date = datetime.datetime.strptime(start_date, "%Y-%m-%d")
    end_date = datetime.datetime.strptime(end_date, "%Y-%m-%d")

    # 计算起始日期所在周的第一天和结束日期所在周的最后一天
    start_week_start = start_date - datetime.timedelta(days=start_date.weekday())
    end_week_end = end_date + datetime.timedelta(days=6 - end_date.weekday())

    # 创建日期范围
    date_range = [start_week_start + datetime.timedelta(days=x) for x in range((end_week_end - start_week_start).days + 1)]

    # 计数器初始化为0
    week_count = 0

    # 遍历日期范围，计算周数
    for date in date_range:
        if date.weekday() == 0:  # 如果是新的一周的开始
            week_count += 1

    return week_count

# 示例用法
start_date = "2022-01-01"
end_date = "2022-12-31"
weeks = count_weeks(start_date, end_date)
print("总周数：", weeks)