曲线拟合python中的分组数据

时间:2017-03-06 01:23:33

标签: python pandas numpy curve-fitting stat

我迫切需要使用我最近开始学习的python解决问题。

我有一个小的趋势数据集(1500行和~30列)。我需要在此数据集上进行多项式曲线拟合,并将新数据集导出为CSV。所以这个数据按2列分组,即。产品和风险和间隔是1-60的时间。我能够一次为1组和1列进行曲线拟合,但无法弄清楚如何对其进行扩展,以便我可以对所有其他列和所有组执行相同的操作。我也是

我已使用以下代码进行此分析。

import pandas as pd
import numpy as np 
import numpy as np
import matplotlib.pyplot as plt
% matplotlib inline

df=pd.read_csv('C:/Users/ashish1104/Desktop/Book1.csv')
df.columns=[x.lower()for x in df.columns]
df.head()

dfg=df.groupby(['product','risk'])

for name, group in dfg:
        x=dfg.get_group(name)['interval']
        y=dfg.get_group(name)['adb']
        z = np.polyfit(x, y, 6)
        f = np.poly1d(z)
        y_new = f(x)
        print (y_new)

如果有人可以帮助解决这个问题我真的很感激。提前谢谢。

Link to dataset

0 个答案:

没有答案