Sympy中IndexedBase对象如何通过列表/数值序列完成批量替换？

IndexedBase本身不支持直接传入Python列表完成批量替换。

原因说明 #

Sympy的IndexedBase是符号化的索引基对象，其subs方法的替换逻辑仅支持符号对象间的匹配映射，你直接传入Python列表时，Sympy无法自动建立x[i]和列表第i个元素的对应关系，因此替换不生效。

可行的批量替换方案 #

你可以选择以下两种常用方法完成替换：

方案1：手动生成替换映射字典 #

遍历索引范围生成每个x[i]到列表对应值的映射规则，直接传入subs即可：

vals = list(range(6))
# 生成替换映射
replace_map = {x[i]: vals[i] for i in range(6)}
# 无需先执行doit()也可完成替换
result = s.subs(replace_map).doit()
# 输出结果为 15，即0+1+2+3+4+5的计算结果

方案2：替换为Sympy原生Array对象 #

Sympy的Array类型支持符号索引匹配，你可以把x替换为Sympy Array即可自动完成索引映射：

from sympy import Array
# 将列表转为Sympy Array
x_sym_arr = Array(range(6))
result = s.subs(x, x_sym_arr).doit()
# 同样得到计算结果 15

注意事项 #

两种方案都需要保证索引范围和你传入的序列长度一致，否则会触发索引越界错误。

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