import numpy as np
import sympy as sym
from IPython.display import display,Math
# 3x^2 + 2x -1 = 0 == 3x^2 + 2x^1 -1x^0 = 0
coefs = [3,2,-1]
roots = np.roots(coefs)
roots
x = sym.symbols('x')
pn = 3*x**2 + 2*x -1
for i in roots:
display(Math('\\text{At }x=%g, \\quad %s = %g' %(i,sym.latex(pn),pn.subs(x,i))))
for i in range(1,11):
coefs = np.arange(1,i+1)
#print(coefs)
print('A degree-%s polynomial has %s roots' %(len(coefs)-1,len(np.roots(coefs))))