Python箭头获取日期范围之间的年份列表

Question

我有这个函数:

def function(start_date_arrow=None,end_date_arrow=None, date_concept=None):
list=[getattr(date, date_concept) for date in arrow.Arrow.range(date_concept, start_date_arrow, end_date_arrow)]

这个函数在迭代date_concept='month'和date_concept='day'时效果很好。另一方面，date_concept='year'只返回一个项目的列表。

例如:

start_date_arrow= arrow.get('2021-11-05')
end_date_arrow= arrow.get('2022-02-05')
year_list=function(start_date_arrow=start_date_arrow,end_date_arrow=end_date_arrow, date_concept='year')
year_list is [2021]   
month_list=function(start_date_arrow=start_date_arrow,end_date_arrow=end_date_arrow, date_concept='month')
month_list is [11, 12, 1, 2]
day_list=function(start_date_arrow=start_date_arrow,end_date_arrow=end_date_arrow, date_concept='day')
day_list is [5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30]

第二个和第三个调用是ok的，但是第一个调用应该返回[2021,2022]而不是[2021]。

你知道年会发生了什么吗?

Answer 1

找到问题。如果使用:

start_date_arrow= arrow.get('2021-11-05')
end_date_arrow= arrow.get('2022-02-05')

两者之间的年份差小于1，因此它只返回第一个年份，因此要在列表中返回2022,end_date_arrow应该是end_date_arrow= arrow.get('2022-11-05')

因此，我使用if语句强制将结束日期大一年，以强制返回这两个年份。