在python中查找两个日期之间的月数

Question

我在熊猫数据框“df”中有一个日期对象和一个日期列“date1”，如下所示：

date = '202107'

df
    date1 
0   2021-07-01   
1   2021-08-01   
2   2021-09-01   
3   2021-10-01   
4   2021-11-01
5   2023-02-01
6   2023-03-01

我想在df中创建一列“月”，其中

months = (date1 + 1month) - date

我的输出数据框应如下所示：

df
    date1        months
0   2021-07-01   1
1   2021-08-01   2
2   2021-09-01   3
3   2021-10-01   4
4   2021-11-01   5
5   2023-02-01   20
6   2023-03-01   21

Answer 1

IIUC

s=(df.date1-pd.to_datetime(date,format='%Y%m'))//np.timedelta64(1, 'M')+1
Out[118]: 
0    1
1    2
2    3
3    4
4    5
Name: date1, dtype: int64
df['months']=s

更新

(df.date1.dt.year*12+df.date1.dt.month)-(pd.to_numeric(date)//100)*12-(pd.to_numeric(date)%100)+1
Out[379]: 
0     1
1     2
2     3
3     4
4     5
5    20
6    21
Name: date1, dtype: int64