Pandas在特定日期创建日期范围

Question

我想创建一个列表（或数组或其他任何东西），每个月间隔一个给定数量的日期。在

基本上我想要的就是这个

>>>some_function(start_date=date(2005, 5, 14), periods=4, freq='M')
['2005-05-14', '2005-06-14', '2005-07-14', '2005-08-14']

如果月初日快到月底了，我要这个

^{pr2}$

我知道pandas date_range函数，但是它会产生这个函数

pd.date_range(date(2005, 5, 14), periods=4, freq='M')
Out[1]: DatetimeIndex(['2005-05-31', '2005-06-30', '2005-07-31', '2005-08-31'],
          dtype='datetime64[ns]', freq='M')

也就是说，它将月末设置为日。这不是我想要的。在

显然，这可以通过迭代周期数产生，但是当startmonth的日期接近该月的最后一天时，这会造成麻烦。在

有没有人知道有一个函数会产生这个函数，或者上面概述的方法是唯一的方法吗？在

Answer 1

这应该行得通。add_months函数通过How to increment datetime by custom months in python without using library中的@davewbb。在

import datetime
import calendar

start_date = '2018-02-02'

def add_months(sourcedate, months):
    month = sourcedate.month - 1 + months
    year = sourcedate.year + month // 12
    month = month % 12 + 1
    day = min(sourcedate.day, calendar.monthrange(year, month)[1])
    return datetime.date(year, month, day)

def range_of_months(sourcedate, months):
    return [add_months(sourcedate, m) for m in range(months+1)]

start = datetime.date.today()

range_of_months(start, 5)

# [datetime.date(2018, 2, 2),
#  datetime.date(2018, 3, 2),
#  datetime.date(2018, 4, 2),
#  datetime.date(2018, 5, 2),
#  datetime.date(2018, 6, 2),
#  datetime.date(2018, 7, 2)]