在pandas中按组添加缺少年份的行

import pandas as pd

# 假设有一个包含日期和值的DataFrame
df = pd.DataFrame({'日期': ['2020-01-01', '2020-01-02', '2020-01-04'],
                   '值': [1, 2, 3]})

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

# 按照日期的年份进行分组
groups = df.groupby(df['日期'].dt.year)

# 自定义操作函数，用于检查并添加缺少的年份行
def add_missing_years(group):
    year = group.name
    start_date = pd.to_datetime(f'{year}-01-01')
    end_date = pd.to_datetime(f'{year}-12-31')
    missing_dates = pd.date_range(start_date, end_date).difference(group['日期'])
    missing_rows = pd.DataFrame({'日期': missing_dates, '值': None})
    return pd.concat([group, missing_rows]).sort_values('日期')

# 在每个组上执行自定义操作
new_df = groups.apply(add_missing_years).reset_index(drop=True)

# 输出结果
print(new_df)