fitting a polynom.py

import numpy as np
import matplotlib.pyplot as plt

plt.rcParams['figure.facecolor'] = '1.0'
plt.rcParams['figure.constrained_layout.use'] = True

x = np.linspace(0, 100)
y = x**3
res = np.polyfit(x, y, deg=5)
f = np.poly1d(res)

fig, ax = plt.subplots()
ax.plot(x, f(x))
ax.scatter(x, y)