如何修改Python代码实现omega从-3到3变化的y=ωx²动画绘制

我想要绘制y = ω*x²的动画，其中ω的取值范围为-3到3，步长0.25；x的取值范围为-4到4，步长0.001。但当前我的代码中omega仅从0开始，请问如何调整代码以实现所需的omega取值范围？

原代码如下：

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()

期望效果如下：

核心问题是原代码中frames=500会默认生成从0开始的整数序列，无法覆盖-3到3的范围。只需手动生成符合要求的ω序列作为动画帧即可，具体修改如下：

1. 生成指定范围的ω序列：用np.linspace(-3, 3, 25)生成从-3到3、步长0.25的25个值（计算方式：(3 - (-3))/0.25 + 1 = 25）
2. 精准生成x序列：将np.linspace(-4, 4, 8000)替换为np.arange(-4, 4.001, 0.001)，严格符合步长0.001的要求（arange是左闭右开区间，需将结束值设为4.001才能包含4）
3. 调整动画速度：可选修改interval参数（比如设为150），让每个ω值的展示更贴合期望效果

修改后的完整代码：

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):
    # 严格按步长0.001生成x取值
    x = np.arange(-4, 4.001, 0.001)
    y = omega * x**2
    line.set_data(x, y)
    return line,

# 生成符合要求的omega帧序列
omega_frames = np.linspace(-3, 3, 25)
anim = FuncAnimation(fig, animate, init_func=init, frames=omega_frames, interval=150, blit=True)

plt.show()

运行修改后的代码，动画会从ω=-3开始，逐步过渡到ω=3，完全匹配你想要的效果。

内容的提问来源于stack exchange，提问作者Ongky Denny Wijaya