Sympy：来自逻辑表达式的 C 代码

Question

从一个 sympy 逻辑表达式，我想得到等效的 C 代码。首先，我注意到您不能使用像 and 和 or 这样的本机逻辑运算符，因为 sympy 会以某种方式将它们剥离。很公平，有& and friends。我试过了

from sympy import *
from sympy.utilities.codegen import codegen

x = Symbol('x')
is_valid = Symbol('is_valid')

# f = x > 0 and is_valid  # TypeError: cannot determine truth value of
f = (x > 0) & is_valid  # And(is_valid, x > 0)

# TypeError: The first argument must be a sympy expression.
[(c_name, c_code), (h_name, c_header)] = codegen(("f", f), "C")

但由于某种原因，我得到了

TypeError: 第一个参数必须是 sympy 表达式。

有什么提示吗？

Answer 1

错误消息基于硬编码的 isinstance 检查。如果它被删除，我得到

#include "f.h"
#include <math.h>

double f(double is_valid, double x) {

   double f_result;
   f_result = is_valid && x > 0;
   return f_result;

}

但是请注意，这可能仍然不是您想要的，因为 is_valid 被设置为双精度，并且您可能希望它是 int（或 C99 bool）。

我的建议：直接在你的表达式上使用ccode，并手动编写函数包装器。如果您需要更具可扩展性的东西，您也可以使用pycodeexport。