numpy.polyfit数据导出到Python pandas.core.frame

时间:2018-08-30 16:40:13

标签: python pandas numpy

我能够在StackOverflow的专家的帮助下开发代码。但是,我陷入了从'numpy.polyfit'函数添加新属性并将数据导出到数据框中的问题。

我的要求:

1。除了坡度外,还添加拦截,右值,p值,stderr(来自ployfit函数)

  1. 将数据导入核心数据帧。

示例代码:

year = [1993, 1994, 1995, 1993, 1994, 1995] 
category =['rev', 'rev', 'exp', 'exp', 'net', 'net'] 
values = [200, 250, 42, 32, 9, 4] 
 df1 = pd.DataFrame({'year': year, 'category': category,'values': values})

        cats = df1.category.unique().tolist()
        slopes = [np.polyfit(df1.loc[df1.category == cat, "year"], df1.loc[df1.category == cat, "values"], 1)[0] for cat in cats]
        for cat, slope in zip(cats, slopes):
            print("{} slope/trend: {:.3f}".format(cat, slope))

非常感谢您。

0 个答案:

没有答案