如何编写一个返回内插值的函数(pandas dataframe)？

import pandas as pd

def interpolate_dataframe(df):
    # 使用interpolate()函数进行内插计算
    interpolated_df = df.interpolate()
    
    return interpolated_df

# 创建一个包含缺失值的dataframe
df = pd.DataFrame({'A': [1, 2, np.nan, 4, 5], 'B': [np.nan, 2, 3, np.nan, 5]})

# 调用函数进行内插计算
interpolated_df = interpolate_dataframe(df)

# 打印内插后的dataframe
print(interpolated_df)

     A    B
0  1.0  NaN
1  2.0  2.0
2  3.0  3.0
3  4.0  4.0
4  5.0  5.0