kafe2_linear_fit_with_xy_data_and_uncertainties.py

1import numpy as np
2from kafe2 import XYFit, Plot
3
4# 1. Define the model function
5def linear_model(x, a=1.0, b=0.0):
6    return a * x + b
7
8# 2. Provide the data
9x_data = [1.0, 2.0, 3.0, 4.0]
10y_data = [1.1, 1.9, 3.2, 3.8]
11y_error = 0.2  # simple absolute error
12
13# 3. Create the Fit object
14fit = XYFit(xy_data=[x_data, y_data], model_function=linear_model)
15fit.add_error(axis='y', err_val=y_error)
16
17# 4. Perform the fit
18fit.do_fit()
19
20# 5. Report the results
21fit.report()
22
23# 6. Plot the results
24plot = Plot(fit)
25plot.plot()
26plot.show()