Pandas:生成两个日期之间的日期间隔，每年重置一次

Question

我正在尝试使用pandas.date_range在两个时间段之间生成8天的间隔。此外，当8天的间隔超过年底(即365/366(时，我希望范围开始重置为相应的年初。下面是两年的示例代码，但是，我确实计划在几年内使用它，例如2014-01-01到2021-01-01。

import pandas as pd
print(pd.date_range(start='2018-12-01', end='2019-01-31', freq='8D'))

结果，

DatetimeIndex(['2018-12-01', '2018-12-09', '2018-12-17', '2018-12-25','2019-01-02', '2019-01-10', '2019-01-18', '2019-01-26'], dtype='datetime64[ns]', freq='8D')

然而，我希望2019年间隔的开始重置为第一天，例如2019-01-01

Answer 1

您可以循环创建一个日期范围，直到每一年的下一年开始，并将其追加到结束日期。

import pandas as pd
from datetime import date
def date_range_with_resets(start, end, freq):
start = date.fromisoformat(start)
end = date.fromisoformat(end)
result = pd.date_range(start=start, end=start, freq=freq) # initialize result with just start date
next_year_start = start.replace(year=start.year+1, month=1, day=1)
while next_year_start < end:
result = result.append(pd.date_range(start=start, end=next_year_start, freq=freq))
start = next_year_start
next_year_start = next_year_start.replace(year=next_year_start.year+1)
result = result.append(pd.date_range(start=start, end=end, freq=freq))
return result[1:] # remove duplicate start date
start = '2018-12-01'
end = '2019-01-31'
date_range_with_resets(start, end, freq='8D')

编辑：这里有一个不使用日期时间的简单方法。在开始和结束之间创建一个年的日期范围，然后循环这些年。

def date_range_with_resets(start, end, freq):
years = pd.date_range(start=start, end=end, freq='YS') # YS=year start
if len(years) == 0:
return pd.date_range(start=start, end=end, freq=freq)
result = pd.date_range(start=start, end=years[0], freq=freq)
for i in range(0, len(years) - 1):
result = result.append(pd.date_range(start=years[i], end=years[i+1], freq=freq))
result = result.append(pd.date_range(start=years[-1], end=end, freq=freq))
return result