Find Coefficients

import numpy as np
import matplotlib.pyplot as plt

x = np.array([1, 2, 3, 4, 4])
y = np.array([2, 4, 6, 8, 9])

print(x)

print(y)

size = np.size(x)

x_mean = np.mean(x)

print(x_mean)

y_mean = np.mean(y)

print(y_mean)

SS_xy = np.sum(y * x - (size * y_mean * x_mean))

print(SS_xy)

SS_xx = np.sum(x * x - (size * x_mean * x_mean))

print(SS_xx)

# Calculate regression coefficients
a1 = SS_xy / SS_xx
b1 = y_mean - (a1 * x_mean)

coeff = b1, a1

print(coeff)