我能够在StackOverflow的专家的帮助下开发代码。但是,我陷入了从'numpy.polyfit'函数添加新属性并将数据导出到数据框中的问题。
我的要求:
1。除了坡度外,还添加拦截,右值,p值,stderr(来自ployfit函数)
示例代码:
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))
非常感谢您。