pandas.tseries.offsets结果不可预测，寻求简化逻辑的解决方案

不用写一堆if判断，有两种简洁可靠的方法实现你要的n==0当月、n==-1上月、n==1下月的预期效果：

方法1：利用Period对象（推荐） #

直接通过日期转周期（Period）的方式，完全规避offsets的歧义行为，结果100%可预测：

月初逻辑 #

import pandas as pd

def get_month_begin(date, n=0):
    # 转成月份周期，加减n后取开始时间
    return (date.to_period('M') + n).start_time

月末逻辑 #

def get_month_end(date, n=0):
    # 转成月份周期，加减n后取结束时间
    return (date.to_period('M') + n).end_time

测试示例

test_dates = [
    pd.Timestamp('2023-05-15'),
    pd.Timestamp('2023-05-31'),
    pd.Timestamp('2023-06-01')
]

for dt in test_dates:
    print(f"原日期: {dt.date()}")
    print(f"上月月初: {get_month_begin(dt, -1).date()}")
    print(f"当月月初: {get_month_begin(dt, 0).date()}")
    print(f"下月月初: {get_month_begin(dt, 1).date()}")
    print(f"上月月末: {get_month_end(dt, -1).date()}")
    print(f"当月月末: {get_month_end(dt, 0).date()}")
    print(f"下月月末: {get_month_end(dt, 1).date()}")
    print("-"*30)

输出结果完全统一，不管输入日期是月中、月末还是月初，都严格对应预期的月份范围。

方法2：封装offsets对象（兼容原有习惯） #

如果一定要用pandas.tseries.offsets，可以先把输入日期锚定到当月的基准点（月初/月末），再进行偏移，避免歧义：

封装后的月初偏移 #

from pandas.tseries.offsets import MonthBegin

def safe_month_begin(date, n=0):
    # 先锚定到当月月初，再偏移n个月
    anchor = date - MonthBegin(n=0)
    return anchor + MonthBegin(n=n)

封装后的月末偏移 #

from pandas.tseries.offsets import MonthEnd

def safe_month_end(date, n=0):
    # 先锚定到当月月末，再偏移n个月
    anchor = date - MonthEnd(n=0)
    return anchor + MonthEnd(n=n)

这个方法通过先消除输入日期的不确定性，再进行偏移，结果同样可预测，但代码可读性不如方法1。

内容的提问来源于stack exchange，提问作者EvanHong