用pycharm求解一阶常微分方程组代码
时间: 2023-07-01 22:13:01 浏览: 170
以下是在PyCharm中使用Python的SciPy库求解一阶常微分方程组的示例代码:
```
import numpy as np
from scipy.integrate import solve_ivp
import matplotlib.pyplot as plt
# 定义微分方程组
def myODE(t, y):
dydt = np.zeros((2,))
dydt[0] = y[1]
dydt[1] = -np.sin(y[0])
return dydt
# 设置初始条件
tspan = (0, 10)
y0 = [1, 0]
# 调用solve_ivp函数求解微分方程组
sol = solve_ivp(myODE, tspan, y0)
# 绘制结果
plt.plot(sol.t, sol.y[0], '-o', label='y1')
plt.plot(sol.t, sol.y[1], '-x', label='y2')
plt.legend()
plt.xlabel('t')
plt.ylabel('y')
plt.show()
```
在PyCharm中,我们可以创建一个新的Python文件并将上述代码粘贴到文件中。然后,我们可以运行该文件以执行代码。
为了绘制结果,我们使用matplotlib库的plot函数和legend函数来绘制y1和y2随时间的变化,并为它们添加一个图例。最后,使用show函数来显示图形。
请确保在PyCharm中正确安装了SciPy和matplotlib库。如果没有安装,可以使用以下命令在PyCharm的终端中安装它们:
```
pip install scipy matplotlib
```
阅读全文