change df，其中日期的年份等于2020?

Question

我有一个列日期的Df，我试图在这个列中搜索年份是2020的地方，并在同一点31天的日期上添加日期，例如Df。日期[1]= 10.12.2020，所以这一行将使用新数据10.1.2021

Answer 1

我想这就是你要找的，如果我理解正确的话:

import datetime
import pandas as pd
# Create a dummy dataframe
date_range = pd.date_range(datetime.datetime(2020, 1, 1), datetime.datetime(2022, 1, 1))
test = pd.DataFrame({"date": date_range})
test

date
0   2020-01-01
1   2020-01-02
2   2020-01-03
3   2020-01-04
4   2020-01-05
..         ...
727 2021-12-28
728 2021-12-29
729 2021-12-30
730 2021-12-31
731 2022-01-01

对于每个值，查找年份是否为2020，如果是，则在日期上加31天:

test['norm_date'] = test['date'].where(test['date'].dt.year != 2020, test['date'].dt.date.add(datetime.timedelta(days=31)))
test

date  norm_date
0   2020-01-01 2020-02-01
1   2020-01-02 2020-02-02
2   2020-01-03 2020-02-03
3   2020-01-04 2020-02-04
4   2020-01-05 2020-02-05
..         ...        ...
727 2021-12-28 2021-12-28
728 2021-12-29 2021-12-29
729 2021-12-30 2021-12-30
730 2021-12-31 2021-12-31
731 2022-01-01 2022-01-01