экспорт данных numpy.polyfit в Python pandas.core.frame - PullRequest
0 голосов
/ 30 августа 2018

Я могу разработать код с помощью экспертов из StackOverflow. Однако я застрял в добавлении новых атрибутов из функции numpy.polyfit и экспорте данных во фрейм данных.

Мои требования:

1.Добавить Intercept, rvalue, pvalue, 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))

Большое спасибо.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...