Python Pandas:从数据fram计算RMSE的简单示例

2024-05-23 17:54:56 发布

您现在位置:Python中文网/ 问答频道 /正文

需要一个用Pandas数据帧计算RMSE的简单示例。如果函数在循环中返回真值和预测值:

def fun (data):
   ...
   return trueVal, predVal

for data in set:
   fun(data)

然后一些代码将这些结果放在以下数据框中,其中x是实际值,p是预测值:

In [20]: d
Out[20]: {'p': [1, 10, 4, 5, 5], 'x': [1, 2, 3, 4, 5]}

In [21]: df = pd.DataFrame(d)

In [22]: df
Out[22]: 
    p  x
0   1  1
1  10  2
2   4  3
3   5  4
4   5  5

问题:

1)如何将fun函数的结果放入df数据帧中?

2)如何使用df数据帧计算RMSE?


Tags: 数据函数in示例pandasdfdatareturn
1条回答
网友
1楼 · 发布于 2024-05-23 17:54:56

问题1
这取决于数据的格式。我希望你已经有了真正的价值,所以这个函数只是一个传递。

问题2

pandas
((df.p - df.x) ** 2).mean() ** .5

numpy
(np.diff(df.values) ** 2).mean() ** .5

相关问题 更多 >