使用matplotlib的Python抛物线拍摄动画

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

# 设置抛物线参数
g = 9.8  # 重力加速度 (m/s^2)
v0 = 10  # 初始速度 (m/s)
theta = np.radians(45)  # 发射角度 (45度)

# 计算抛物线轨迹
t_max = 2 * v0 * np.sin(theta) / g
t_values = np.linspace(0, t_max, num=100)
x_values = v0 * np.cos(theta) * t_values
y_values = v0 * np.sin(theta) * t_values - 0.5 * g * t_values**2

# 创建画布和轴
fig, ax = plt.subplots()
line, = ax.plot([], [], lw=2)

# 初始化函数
def init():
    ax.set_xlim(0, x_values.max())
    ax.set_ylim(0, y_values.max())
    return line,

# 动画更新函数
def update(frame):
    line.set_data(x_values[:frame], y_values[:frame])
    return line,

# 创建动画
ani = FuncAnimation(fig, update, frames=len(t_values), init_func=init, blit=True)

# 显示动画
plt.show()