micropython-ulab/tests/poly.py
2020-10-17 21:26:59 +02:00

25 lines
419 B
Python

import ulab as np
# polynom evaluation
x = np.linspace(0, 10, num=9)
p = [1, 2, 3]
y = np.polyval(p, x)
print(y)
# linear fit
x = np.linspace(-5, 5, num=11)
y = x + np.sin(x)
p = np.polyfit(x, y, 1)
print(p)
# quadratic fit
x = np.linspace(-5, 5, num=11)
y = x*x + np.sin(x)*3.0
p = np.polyfit(x, y, 2)
print(p)
# cubic fit
x = np.linspace(-5, 5, num=11)
y = x*x*x + np.sin(x)*10.0
p = np.polyfit(x, y, 3)
print(p)