python - 在Python中将年份添加到日期(d-m-y)列

Question

我有一个数据表，其中 Col1 的日期格式为 (2019-08-19)，另一个 Col2 的年份格式为(1 表示 1 年，2 表示 2 年...)，目标是将 col1 添加到将第 2 列更改为新列，结果为 2020-08-19。

我做了 if 函数，但很长，我想要简单易行的方法

Answer 1

您需要使用.apply与 Dateoffset .

df.apply(lambda x: x['col1'] + pd.DateOffset(years=x['col2']), axis=1)

最小示例:

df = pd.DataFrame({
    'col1': ['2019-08-19', '2019-04-25'],
    'col2': [1,2]
})
df['col1'] = pd.to_datetime(df['col1'])
df['col1'] = df.apply(lambda x: x['col1'] + pd.DateOffset(years=x['col2']), axis=1)

输出:

0   2020-08-19
1   2021-04-25
dtype: datetime64[ns]