为系列中的每一行添加月数

Question

我有一个带有日期列orig_date的pandas数据帧（fb）。我想将count_to_add列中的月份数添加到orig_date列，但MonthEnd函数未向量化。

这是我试过的：

fb["orig_date"] = fb["orig_date"] + 
                  pd.tseries.offsets.MonthEnd(fb["count_to_add"])

由于count_to_add系列不是整数，因此MonthEnd函数出错。

Answer 1

如果您只想添加“月数”，请改用DateOffset。

fb['orig_date'] = fb.apply(lambda x: x['orig_date'] + pd.tseries.offsets.DateOffset(months=int(x['count_to_add'])), axis=1)