熊猫：根据int值改变年份

Question

我试图根据另一列中的数字从一列中减去年数。

这就是我的意思：

   base_date          amount_years
0  2006-09-01         2
1  2007-04-01         4

结果将是：

   base_date          amount_years
0  2008-09-01         2
1  20011-04-01        4

有没有办法在python中实现这个目标？

Answer 1

对apply和axis=1使用DateOffset进行每行处理：

f = lambda x: x['base_date'] + pd.offsets.DateOffset(years=x['amount_years'])
df['base_date'] = df.apply(f, axis=1)
print (df)
   base_date  amount_years
0 2008-09-01             2
1 2011-04-01             4