我想用mutplotlib绘制物理平抛运动小球的运动轨迹动画，我需要在顶层窗口中输入水平速度和竖直高度，就能实现平抛运动的演示

好的，您可以使用matplotlib库中的animation模块来绘制平抛运动小球的运动轨迹动画。下面是一个简单的示例代码，可以帮助您实现这个功能：

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

# 定义重力加速度
g = 9.8

# 定义动画帧数
frames = 100

# 定义时间间隔
dt = 0.1

# 获取用户输入的竖直高度和水平速度
h = float(input("请输入竖直高度（单位：米）："))
v0 = float(input("请输入水平速度（单位：米/秒）："))

# 计算小球的运动轨迹
t = np.linspace(0, np.sqrt(2*h/g), frames)
x = v0*t
y = h - 0.5*g*t**2

# 绘制动画
fig, ax = plt.subplots()
ax.set_xlim(0, np.max(x))
ax.set_ylim(0, np.max(y))
ball, = ax.plot([], [], 'o', color='red')

def animate(i):
    ball.set_data(x[:i], y[:i])
    return ball,

ani = animation.FuncAnimation(fig, animate, frames, interval=dt*1000, blit=True)

# 显示动画
plt.show()

在这个示例代码中，我们首先定义了重力加速度、动画帧数和时间间隔。然后获取用户输入的竖直高度和水平速度，并计算小球的运动轨迹。最后使用matplotlib库中的animation模块来绘制动画。

在动画绘制时，我们使用了animate函数来更新小球的位置，使用FuncAnimation函数来生成动画。最后调用show函数来显示动画。

这个示例代码只是一个简单的示例，您可以根据自己的需求进行修改和扩展。