如何用参数构建含常微分方程项的NumPy矩阵

你原来的代码出错是因为np.zeros默认创建浮点型数组，只能存储数值，无法直接赋值字符串类型的参数。下面提供两种可行方案：

方案一：使用numpy存储字符串形式的参数 #

创建支持存储任意对象的numpy数组，手动填充指定位置的字符串参数：

import numpy as np

# 创建5×5的object类型数组（可存储字符串、表达式等）
mat = np.empty((5, 5), dtype=object)
# 先将所有位置填充为空字符串（可选，按需调整）
mat[:] = ''

params = ['-lambda', 'lambda', '-u-v', 'u', '-w', 'v', 'w','-gamma', 'gamma']
pos = [(0,0),(1,0),(1,1),(2,1),(2,2),(3,1),(3,2),(3,3),(4,3)]

# 逐个位置赋值参数
for param, (row, col) in zip(params, pos):
    mat[row, col] = param

# 查看结果
print(mat)

方案二：使用SymPy构建符号矩阵（推荐用于后续ODE运算） #

如果最终要实现矩阵与向量相乘得到ODE表达式，用符号计算库SymPy更高效——它能直接处理符号运算，无需手动解析字符串：

import sympy as sp

# 定义所有符号参数
lambda_sym, u, v, w, gamma = sp.symbols('lambda u v w gamma')
# 定义状态向量（替换为你实际的ODE状态变量，比如U, X, Y, Z, W等）
state_vars = sp.Matrix(['U', 'X', 'Y', 'Z', 'W'])

# 初始化5×5零矩阵
mat = sp.zeros(5, 5)

# 按位置填充符号表达式
mat[0, 0] = -lambda_sym
mat[1, 0] = lambda_sym
mat[1, 1] = -u - v
mat[2, 1] = u
mat[2, 2] = -w
mat[3, 1] = v
mat[3, 2] = w
mat[3, 3] = -gamma
mat[4, 3] = gamma

# 计算矩阵与状态向量的乘积，得到各ODE的右端项
ode_equations = mat * state_vars
print(ode_equations)

运行后会直接输出符号化的ODE表达式，比如第一行结果对应dU/dt的表达式，完全符合你的需求。

内容的提问来源于stack exchange，提问作者Landon