在Pandas中迭代计算月数

import pandas as pd

# 创建示例 DataFrame
data = {
    'start_date': ['2020-01-15', '2020-03-20', '2020-05-10'],
    'end_date': ['2020-02-10', '2020-04-15', '2020-06-05']
}
df = pd.DataFrame(data)

# 将日期列转换为 datetime 类型
df['start_date'] = pd.to_datetime(df['start_date'])
df['end_date'] = pd.to_datetime(df['end_date'])

# 计算月数差
df['months_diff'] = (df['end_date'].dt.year - df['start_date'].dt.year) * 12 + df['end_date'].dt.month - df['start_date'].dt.month

print(df)

  start_date   end_date  months_diff
0 2020-01-15 2020-02-10           0
1 2020-03-20 2020-04-15           1
2 2020-05-10 2020-06-05           1

df['start_date'] = pd.to_datetime(df['start_date'], format='%Y-%m-%d')
df['end_date'] = pd.to_datetime(df['end_date'], format='%Y-%m-%d')

from dateutil.relativedelta import relativedelta

df['months_diff'] = df.apply(lambda row: (relativedelta(row['end_date'], row['start_date']).years * 12 + relativedelta(row['end_date'], row['start_date']).months), axis=1)