以下是一个示例代码，用于按月分组计算日期差。

from datetime import datetime

def calculate_month_diff(start_date, end_date):
    # 计算起始日期和结束日期的年份和月份
    start_year = start_date.year
    start_month = start_date.month
    end_year = end_date.year
    end_month = end_date.month

    # 计算起始日期和结束日期之间的月份数
    month_diff = (end_year - start_year) * 12 + (end_month - start_month)

    # 如果结束日期的日大于起始日期的日，则月份数加1
    if end_date.day > start_date.day:
        month_diff += 1

    return month_diff

# 测试代码
start_date = datetime(2022, 1, 15)
end_date = datetime(2022, 4, 20)
diff = calculate_month_diff(start_date, end_date)
print(f"起始日期：{start_date}")
print(f"结束日期：{end_date}")
print(f"跨越了 {diff} 个月")

这个示例代码中，calculate_month_diff 函数接受起始日期和结束日期作为参数，然后计算它们之间的月份数。具体的计算逻辑如下：

1. 首先，获取起始日期和结束日期的年份和月份。
2. 接着，计算起始日期和结束日期之间的月份数。这里使用了一个简单的公式：(end_year - start_year) * 12 + (end_month - start_month)。
3. 如果结束日期的日大于起始日期的日，则月份数加1。这是为了处理例如起始日期是1月31日，结束日期是2月28日的情况，按照月份计算，实际上跨越了两个月。

最后，我们使用示例数据进行测试，并打印出结果。输出应该是：

起始日期：2022-01-15 00:00:00
结束日期：2022-04-20 00:00:00
跨越了 4 个月

你可以根据需要对这个示例代码进行修改和扩展，以适应不同的需求。