python绘制倒立摆代码
时间: 2023-10-11 19:14:28 浏览: 60
python圣诞树编写实例详解
我可以为您提供绘制倒立摆的Python代码,您可以参考以下代码:
```
import numpy as np
import matplotlib.pyplot as plt
from scipy.integrate import odeint
# 倒立摆的运动方程
def pend(y, t, b, c):
theta, omega = y
dydt = [omega, -b*omega - c*np.sin(theta)]
return dydt
# 参数设定
b = 0.25
c = 5.0
y0 = [np.pi - 0.1, 0.0]
t = np.linspace(0, 10, 101)
# 解微分方程并绘制图像
sol = odeint(pend, y0, t, args=(b, c))
plt.plot(t, sol[:, 0], 'b', label='theta(t)')
plt.plot(t, sol[:, 1], 'g', label='omega(t)')
plt.legend(loc='best')
plt.xlabel('t')
plt.grid()
plt.show()
```
希望以上代码可以帮到您。
阅读全文