python的datetime模块获取两个日期之间的月份数，打印两个日期之间所有的年月

获取两个日期之间的月份数

from datetime import datetime

start_date = datetime.strptime('2021-01', '%Y-%m')
start_year, start_month = start_date.year, start_date.month
end_date = datetime.strptime('2021-09', '%Y-%m')
end_year, end_month = end_date.year, end_date.month
interval=(end_year - start_year)*12 + (end_month - start_month)
print(interval)

输出：8

打印两个日期之间所有的年月

start_date = datetime.strptime('2021-01', '%Y-%m')
start_year, start_month = start_date.year, start_date.month
end_date = datetime.strptime('2021-09', '%Y-%m')
end_year, end_month = end_date.year, end_date.month
interval=(end_year - start_year)*12 + (end_month - start_month)
print(interval)

cur_ym = []
for i in range(interval):
    now_month = start_month + i
    year_tmp, month_tmp = start_year + now_month // 12, now_month % 12
    if month_tmp == 0:
        month_tmp = 12
        year_tmp -= 1
    cur_date = datetime.strptime(f'{year_tmp}-{month_tmp}', '%Y-%m')
    interval_date = cur_date.strftime('%Y-%m')
    cur_ym.append(interval_date)
print(cur_ym)

输出：['2021-01', '2021-02', '2021-03', '2021-04', '2021-05', '2021-06', '2021-07', '2021-08']