Function 函数中的均方误差输出包括数据类型和'；0'；

我想在函数中计算fb prophet forecast的测试统计数据，因为我想在使用fb prophet cross_验证获得df_cv后，平均不同预测和截止点的测试统计数据。我创建了一个函数，在按截止点分组后应用于数据帧，以便接收每个截止点的度量值。然后我计算所有这些值的平均值

问题在于，我的函数不仅返回我要查找的值，还返回0以及数据类型的信息。我仍然可以用返回的值进行计算，但当我想绘图等时，这是非常不方便的。如何从输出中去除这些不必要的值

def compute_avg_stats(df_cv,perf_measure):
    measures = {'mse':mean_squared_error,'mae':mean_absolute_error,'mape':mean_absolute_percentage_error,'rmse':mean_squared_error}
    performance_stats = {}
    if perf_measure == 'rmse':
        measure = np.sqrt(measures[perf_measure](y_true=df_cv['y'],y_pred=df_cv['yhat']))
    else:
        measure = measures[perf_measure](y_true=df_cv['yu'],y_pred=df_cv['yhat'])

    return measure

---

我认为

.mean（）

返回一个序列。尝试使用

.mean（）[0]

就像宝石一样工作！这么简单，但我想不出来。非常感谢你！

df_cv.groupby('cutoff').apply(compute_avg_stats,perf_measure='rmse').to_frame().mean()