Я могу разработать код с помощью экспертов из StackOverflow. Однако я застрял в добавлении новых атрибутов из функции numpy.polyfit и экспорте данных во фрейм данных.
Мои требования:
1.Добавить Intercept, rvalue, pvalue, 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))
Большое спасибо.