Блог

Я создал программу, которая строит многочлены, которые вы можете увидеть ниже:

 def symbolic_derivative(xs):

    derivative_xs = []
    derivative_copy = xs.copy()

    if len(xs) == 0:
        return []

    derivative_copy.pop(0)

    for degree, term in enumerate(derivative_copy, start = 1):
        derivative_xs.append(term*degree)
    
    return derivative_xs

def eval_poly(p,x): 
    ans = 0
    for degree,coeff in enumerate(p): # vi använder oss av en for loop och enumerate för att kunna ta fram polynomets värde i en viss punkt x
        ans  = coeff*x**degree # ekvationen som ska räknas
    return ans

import matplotlib.pyplot as plt

def plot_poly(p,x_start=-10,x_end=10,color='b'):

    lst_x = list(range(x_start, x_end   1))

    lst_y = [eval_poly(p,x) for x in lst_x]    

    plt.plot(lst_x, lst_y, color)
    plt.savefig('plot1.pdf')
    plt.show()
 

Затем я хочу создать два графика plot_poly([3,-1,2,1],-10,10,'b') , plot_poly(symbolic_derivative([3,-1,2,1]),-10,10,'r') и я делаю это, вводя их по одному в поле консоли, а затем он выводит график для соответствующей команды.

Но я хотел бы, чтобы python отображал их на том же рисунке, так что после того, как я создал первый график и он отобразился, я бы хотел, чтобы он продолжал рисовать второй график на первом рисунке, а не на новом. То есть python должен добавить график на том же рисунке, что и предыдущий. Как я могу это сделать?

Вот как я хочу работать:

>>> plot_poly([3,-1,2,1],-10,10,'b')

>>> plot_poly(symbolic_derivative([3,-1,2,1]),-10,10,'r')