请写一个画图后坠落的物理引擎python代码

以下是一个简单的画图后坠落的物理引擎 Python 代码：

import matplotlib.pyplot as plt
import numpy as np

# 设置常数
g = 9.8  # 重力加速度
t = 0.0  # 初始时间
dt = 0.01  # 时间步长

# 设置初始条件
v0 = 10.0  # 初始速度
y0 = 0.0  # 初始高度
y = y0  # 当前高度
v = v0  # 当前速度

# 初始化数组
ys = [y0]
vs = [v0]
ts = [t]

# 循环计算位置和速度
while y >= 0.0:
    a = -g  # 加速度
    v = v + a * dt  # 计算速度
    y = y + v * dt  # 计算高度
    t = t + dt  # 更新时间
    ys.append(y)
    vs.append(v)
    ts.append(t)

# 绘制图形
plt.plot(ts, ys)
plt.xlabel('Time (s)')
plt.ylabel('Height (m)')
plt.title('Projectile Motion')
plt.show()

这段代码使用欧拉方法数值积分计算了物体的坠落轨迹，并使用 Matplotlib 库绘制了高度随时间变化的图形。可以根据需要调整参数和初始条件来模拟不同的情况。