diff --git a/scripts/linear_fit.py b/scripts/linear_fit.py new file mode 100644 index 0000000..a382087 --- /dev/null +++ b/scripts/linear_fit.py @@ -0,0 +1,26 @@ +import numpy as np +from scipy.stats import linregress +import matplotlib.pyplot as plt + +# Generate some example data +x = np.array([1, 2, 3, 4, 5]) +y = np.array([2.5, 3.5, 4.5, 5.5, 6.5]) + +# Perform linear regression +slope, intercept, r_value, p_value, std_err = linregress(x, y) + +# Calculate R^2 +r_squared = r_value**2 + +# Print the slope, intercept, and R^2 +print("Slope:", slope) +print("Intercept:", intercept) +print("R^2:", r_squared) + +# Plot the data and the linear fit +plt.scatter(x, y, label='Data') +plt.plot(x, slope * x + intercept, color='red', label='Linear Fit') +plt.xlabel('X') +plt.ylabel('Y') +plt.legend() +plt.show()