pandasで線形回帰をする

pandasで線形回帰をするにはols関数を使えばいけますが、statsmodelsという特殊なパッケージをインストールしていないといけません

statsmodelsをインストールしていないと

ImportError: No module named scikits.statsmodels.api

みたいなエラーがでます。

それを踏まえて線形回帰をするとこのような感じ

import pandas as pd
import matplotlib.pyplot as plt
plt.style.use('ggplot')

df_origin = #適当な系列データ
linear_model = pd.ols(y=df_origin['売上'],x=df_origin['時間'])
df_test = pd.DataFrame({
    'origin':df_origin['売上'],
    'regresssion':linear_model.y_fitted
},index=df_origin['時間'])
df_test.plot()

コメントを残す

メールアドレスが公開されることはありません。 * が付いている欄は必須項目です

このサイトはスパムを低減するために Akismet を使っています。コメントデータの処理方法の詳細はこちらをご覧ください