python - 如何基于变量更改表达式？

Question

我正在使用 SymPy coeff 功能，并希望根据变量(维度)调整我的表达式。

如果维度等于 2，则

y = expr.coeff(v[0],0).coeff(v[1],0)

如果维度等于 3，则

y = expr.coeff(v[0],0).coeff(v[1],0).coeff(v[2],0)

我如何概括这一点以允许任意维度？

Answer 1

迭代维度的数量。我们称之为dim

y = expr
for i in range(dim):
    y = y.coeff(v[i], 0)

这会沿着系数列表“走”下去，一次一个。