diff --git a/pymbolic/interop/common.py b/pymbolic/interop/common.py index b0b73e5374c1bece0689bae6e79a6f4ef8f1faa4..7501a5078f3e7c52badcef4ba36c543275426c0d 100644 --- a/pymbolic/interop/common.py +++ b/pymbolic/interop/common.py @@ -144,6 +144,12 @@ class PymbolicToSympyLikeMapper(EvaluationMapper): else: self.raise_conversion_error(expr) + def map_subscript(self, expr): + if isinstance(expr.aggregate, prim.Variable) and isinstance(expr.index, int): + return self.sym.Symbol("%s_%d" % (expr.aggregate.name, expr.index)) + else: + self.raise_conversion_error(expr) + def map_substitution(self, expr): return self.sym.Subs(self.rec(expr.child), tuple(self.sym.Symbol(v) for v in expr.variables),