이제 최소 제곱근 공식으로 기울기 a의 값과 y 절편 b의 값을 구해 보겠습니다.

x의 모든 원소 평균을 구하는 넘파이 함수는 mean()입니다. mx 변수에는 x 원소들의 평균값을, my 변수에는 y 원소들의 평균값을 넣습니다.

mx = np.mean(x)
my = np.mean(y)

이제 앞서 살펴본 최소 제곱근 공식 중 분모 값, 즉 ‘x의 각 원소와 x의 평균값들의 차를 제곱하라’는 파이썬 명령을 만들 차례입니다. 다음과 같이 divisor라는 변수를 만들어 구현할 수 있습니다.

divisor = ➊sum([(i - mx)➋**2 ➌for i in x])

이제 분자에 해당하는 부분을 구하겠습니다. x와 y의 편차를 곱해서 합한 값을 구하면 됩니다. 다음과 같이 새로운 함수를 정의해서 dividend 변수에 분자 값을 저장합니다.

def top(x, mx, y, my):
    d = 0
    for i in range(len(x)):
        d += (x[i] - mx) * (y[i] - my)
    return d
dividend = top(x, mx, y, my)