python - 为 sympy 表达式 "e1 op e2"生成 C 代码，其中 op 是算术比较运算符

Question

假设我有一个sympy 表达式

e1 op e2

哪里op是任何算术比较运算符，例如 <, >, <=等

我要生成C代码

double f(double x1,double x2,..., double xn){
  if (e1 op e2) return 0 else return 1;
}

其中 x1,...,x2 是 e1 或 e2 中的自由变量。我可以使用 sympy.printing.ccode 或 sympy.utlilities.codegen 来简化此代码生成吗？

Answer 1

这是一个使用 ccode 的例子和 Piecewise (您需要提供函数体作为模板):

>>> from sympy import symbols, Piecewise, ccode
>>> x, y, z = symbols('x y z')
>>> e1 = x + 2
>>> e2 = y - 5
>>> rel = e1 < e2
>>> pw = Piecewise((0, rel), (1, True))
>>> print(ccode(pw, assign_to=z))
if (x + 2 < y - 5) {
   z = 0;
}
else {
   z = 1;
}