使用Plotly在单个图形中绘制多个ODE的多个图

pip install plotly

import numpy as np
import plotly.graph_objects as go
from scipy.integrate import odeint

# 定义多个ODE函数
def ode1(y, t):
    return -2 * y

def ode2(y, t):
    return y - t**2

# 定义初始条件
initial_conditions = [1.0, 2.0]

# 定义时间范围
t_span = (0, 5)
t_eval = np.linspace(*t_span, 100)

# 解多个ODE
solutions1 = odeint(ode1, initial_conditions[0], t_eval)
solutions2 = odeint(ode2, initial_conditions[1], t_eval)

# 创建Plotly图形
fig = go.Figure()

# 添加多个ODE曲线到图中
fig.add_trace(go.Scatter(x=t_eval, y=solutions1, mode='lines', name='ODE 1'))
fig.add_trace(go.Scatter(x=t_eval, y=solutions2, mode='lines', name='ODE 2'))

# 更新布局
fig.update_layout(title='Multiple ODE Curves in a Single Plot',
                  xaxis_title='Time',
                  yaxis_title='Solution')

# 显示图形
fig.show()