Построение параболы с входами в matplotlib

Question

, поэтому я создал решатель квадратов c в python, и мне нужен график для go вместе с ним. Я хочу, чтобы этот график изображал параболу.

a = вход для квадратичного c уравнения

b = вход для квадратичного c уравнения

c = вход для c в квадратах c уравнение

y = -b / 2 * a (вектор)

import math
a = int(input('Input a: '))
b = int(input('Input b: '))
c = int(input('Input c: '))
discrim = int((b*b) - 4*a*c) 
posi = (-b + (math.sqrt(discrim)))/(2*a)
neg = (-b - (math.sqrt(discrim)))/(2*a)


if b > 0 and c > 0 and posi < 0:
  posi = -posi
  if neg < 0:
    neg = -neg

y = -b/2*a
vector = (neg+posi)/2, y)

С этим кодом у меня есть мои x intercept (s) и вектор , что достаточно для создания параболы, я просто не уверен, как превратить эти значения в график с помощью MatPlotLib. Пожалуйста, задавайте любые вопросы.

Answer 1

Я внес некоторые исправления в Ваш код. Ниже кода будет построена любая парабола. Вы можете добавить ноль и крайнюю точку на график.

import matplotlib.pyplot as plt
import math
import numpy as np

a = int(input('Input a: '))
b = int(input('Input b: '))
c = int(input('Input c: '))

# calculate delta and zero points
delta = b**2 - 4*a*c
if delta > 0:
    x_1 = (-b + math.sqrt(delta))/(2*a)
    x_2 = (-b - math.sqrt(delta))/(2*a)
if delta == 0:
    x_0 = -b/(2*a)
else:
    pass

# calculate parabola extreme coordinates
p = -b/(2*a)
q = -delta/(4*a)
extreme = [p,q]

# define parabola function
def parabola(x,a,b,c):
    y = a*x**2 + b*x + c
    return y

# plot function
x = np.linspace(int(p)-5,int(p)+5,100)
y = parabola(x,a,b,c)
plt.plot(x,y)
plt.axhline(y=0, color='black', linestyle='-')
plt.axvline(x=0, color='black', linestyle='-')
plt.text(p-0.5, q-3, '[' + str(round(p,2)) +',' + str(round(q,2)) + ']',color='orange', fontsize=9)
plt.plot(p, q, marker="o")

if delta > 0:
    plt.plot(x_1, 0, marker="o", color='green')
    plt.text(x_1 - 0.5, 2, '[' + str(round(x_1,2)) + ']', color='green', fontsize=9)
    plt.plot(x_2, 0, marker="o", color='green')
    plt.text(x_2 - 0.5, 2, '[' + str(round(x_2,2)) + ']', color='green', fontsize=9)

if delta == 0:
    plt.plot(x_0, 0, marker="o", color='green')
    plt.text(x_0 - 0.5, 2, '[' + str(round(x_0,2)) + ']', color='green', fontsize=9)

plt.show()

Пример выборки: