criterio de routh

criterio_de_routh.py

import numpy as np
from math import ceil
from sympy import Symbol
from prettytable import PrettyTable

# [a0, a1, a2, ...]
k = Symbol('k')
# coef = [1, 2, 3, 4, 5]
coef = [1, 8, 32, k]

base = [[0] * len(coef) for _ in range(len(coef))]

for i, a in enumerate(coef):
    base[i%2][i//2] = a/1

for i in range(2, len(coef)):
    for j in range(len(coef)-1):
        # print(f"i={i}, j={j}, {base[i-1][0]}*{base[0][j+1]} - {base[i-2][0]}*{base[i-1][j+1]}/{base[i-1][0]}")
        base[i][j] = (base[i-1][0]*base[i-2][j+1] - base[i-2][0]*base[i-1][j+1])/base[i-1][0]

p = PrettyTable()
for row in base:
    p.add_row(row)

print(p.get_string(header=False, border=False))