Анимация для построения графика y=omega*x^2, где omega варьируется от -3 до 3. ⇐ Python

[Anonymous]

Я хочу построить график y = omega*x^2, где omega варьируется от -3 до 3 с размером шага 0,25 (и x охватывает от -4 до 4 с размером шага 0,001). Текущая версия моего кода (ниже) имеет начало омеги только с 0, а не с -3. Как мне настроить свой код, чтобы омега-значение изменялось в нужном мне диапазоне?

Код: Выделить всё

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.animation import FuncAnimation

fig = plt.figure()
axis = plt.axes(xlim=(-4, 4), ylim=(-40, 40))
line, = axis.plot([], [], lw=3)

def init():
line.set_data([], [])
return line,

def animate(omega):
x = np.linspace(-4, 4, 8000)
y = omega*x**2
line.set_data(x, y)
return line,

anim = FuncAnimation(fig, animate, init_func=init, frames=500, interval=100, blit=True)

plt.show()

Это мой желаемый результат:



Подробнее здесь: https://stackoverflow.com/questions/786 ... rom-3-to-3