【解説記事】Pythonのnumpy polyfitを使って1次関数の係数を求めよう！

PythonのNumPyパッケージには、多項式回帰の計算を行うためのpolyfit関数が用意されています。1次回帰を行う場合は、polyfit関数にx座標のデータ、y座標のデータ、そして回帰の次数を指定して実行します。

具体的には、以下のように実行します。

import numpy as np

# データの準備
x = np.array([1, 2, 3, 4, 5])
y = np.array([2, 3, 5, 6, 7])

# 1次回帰を行う
z = np.polyfit(x, y, 1)

# 回帰直線の傾きと切片を取得する
slope, intercept = z

# 回帰直線の式を表示する
print("y = {:.2f}x + {:.2f}".format(slope, intercept))

このコードでは、xとyの配列を用意して、np.polyfit関数に渡しています。次数には1を指定し、回帰直線の傾きと切片を取得します。最後に、傾きと切片を使って、回帰直線の式を表示しています。

上記のコードを実行すると、以下のような出力が得られます。

y = 1.30x + 0.7

この式は、回帰直線の方程式を表しており、xの値を代入することでyの予測値を得ることができます。