← retour aux snippets

numpy: polyfit & polyval

Ajuster un polynôme et évaluer la courbe.

python numpy #numpy#polyfit#polyval

objectif

Ajuster un polynôme et évaluer la courbe.

code minimal

import numpy as np
x = np.arange(5); y = x**2
coeff = np.polyfit(x, y, deg=2)
print(len(coeff) == 3)

utilisation

import numpy as np
print(np.polyval([1,0,0], 2).item())

variante(s) utile(s)

import numpy as np
p = np.poly1d([1,0])
print(p([1,2]).tolist())

notes

  • Sur de grands degrés, préférez régularisation.