Калькулятор многочленов. Задание переписать код

from numpy.polynomial import Polynomial as P
import numpy as np
 
 
def multiply_polys(A, B, m, n):
    prod = [0] * (m + n - 1);
 
    for i in range(m):
        for j in range(n):
            prod[i + j] += A[i] * B[j];
 
    return prod;
 
def polyn_from_string():
    x = P([0, 1])
 
    P1 = input("Enter polynomial:\n")#-0.867*x**23 + 2.45*x**23 - 64.654*x**11 + 356.897*x**24 + 15864.5465*x
 
    iksy = [pos for pos, char in enumerate(P1) if char == 'x']
    temp = 0
 
    for i in iksy:
        P1 = P1[:i+temp] + '*'+ P1[i+temp:]
        temp += 1
 
    P1 = P1.replace("^", "**")
    P1 = eval(P1)
    return P1
 
def print_poly(P1):
    coefs = np.array(P1.coef)
    nonzerocoefs = coefs[np.nonzero(coefs)]
 
    out = ""
 
    for i in nonzerocoefs:
        if np.where(nonzerocoefs == i)[0][0] == 0:
            if np.where(coefs == i)[0][0] > 1:
                out += "{0}x^{1}".format(i, np.where(coefs == i)[0][0])
            else:
                out += "{0}x".format(i)
        else:
            if i > 0:
                out += " + {0}x^{1}".format(i, np.where(coefs == i)[0][0])
            else:
                out += " - {0}x^{1}".format(-1 * i, np.where(coefs == i)[0][0])
 
    print("out: ", out)
 
P1 = polyn_from_string() #-0.867x^23 + 2.45x^23 - 64.654x^11 + 356.897x^24 + 15864.5465x
# print(P1.coef)
# input()
 
while True:
    choice = input("Choose operation:\n1 - Addition\n2 - Substraction\n3 - Multiplication\n4 - Exit\n")
    if int(choice) == 1:
        P2 = polyn_from_string()
        if len(P1) > len(P2):
            P3 = np.add( np.array(P1.coef), np.hstack([np.array(P2.coef), np.zeros([len(P1)-len(P2)])]) )
        else:
            P3 = np.add( np.hstack([np.array(P1.coef), np.zeros([len(P2)-len(P1)])]), np.array(P2.coef) )
        print_poly(P(P3))
    elif int(choice) == 2:
        P2 = polyn_from_string()
        if len(P1) > len(P2):
            P3 = np.add(np.array(P1.coef), np.hstack([-1*np.array(P2.coef), np.zeros([len(P1) - len(P2)])]))
        else:
            P3 = np.add(np.hstack([np.array(P1.coef), np.zeros([len(P2) - len(P1)])]), -1*np.array(P2.coef))
        print_poly(P(P3))
    elif int(choice) == 3:
        P2 = polyn_from_string()
        P3 = np.polymul(np.poly1d(P1.coef), np.poly1d(P2.coef))
        P3 = multiply_polys(P1.coef, P2.coef, len(P1), len(P2))
        print_poly(P(P3))
    elif int(choice) == 4:
        exit()