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()

Leave a Reply

Your email address will not be published. Required fields are marked *

This site uses Akismet to reduce spam. Learn how your comment data is processed.