polinomio, ricavare un punto e ricavare i punti equidistanti in python

#esercizio1
import numpy as np
#polinomio
def poli(x):
return (x-1)**9

def punti_equidistanti(a, b, n):
x = np.linspace(a, b, n+1)
return x

#intervallo a
a = 0
#intervallo b
b = 2
#n
n = 9

val = poli(a)

#solo con una x e una y
import matplotlib.pyplot as plt
plt.plot(a, val, 'o')
plt.show()

pti = punti_equidistanti(a, b, n)
print("\nPunti equidistanti assegnati a una matrice ")
PU = np.array(pti)
print(PU)
#nel polinomio, gli sto passando 10 punti e non solo uno
plt.plot(pti, poli(pti), 'o')
plt.show()