如何解决Sympy - 生成 C 代码将 Rational 转换为浮动
我正在尝试通过函数 ccode 生成带有 sympy 的 C 代码。
目前,我正在尝试将 sympy 的有理数转换为浮点数以加快计算速度。
例如我有:
import sympy as sp
x=sp.Symbol('x')
y=sp.Symbol('y')
d=sp.Symbol('d')
test=sp.Matrix([
[x/3 + y + 2*d/3,0],[0,x/3 + y + 2*d/3,x/3 + y - 2*d/3]])
res = sp.cse(test)
lines = []
for i,result in enumerate(res):
lines.append(sp.ccode(result,"result_%i"%i,user_functions={'Mul':[(lambda x: x.args[0].is_Rational,lambda x: sp.N(x,n=17))]}))
如果 res
的矩阵部分有一个分数 (res[1]
),我可以用一个 for 循环和一个 try/except 语句来循环它:
for i in range(len(res[1])):
try:
res[1][i].args[0].is_Rational
except:
continue
else:
res[1][i]=sp.N(res[0][i],n=20)
并使用函数 sp.N
将有理值更改为浮点数。
但是,我很难定义一个 lambda 函数/任何其他函数来对 res[0]
中的元组列表执行此操作。
希望得到一些帮助!
亲切的问候
解决方法
您可以采用与 your previous question 类似的方法:子类化代码生成器并覆盖 _print_Rational()
。可以通过 user_functions
参数添加自定义 settings=
:
import sympy as sp
from sympy.printing.c import C99CodePrinter
class CustomCodePrinter(C99CodePrinter):
def _print_Rational(self,expr):
return str(sp.N(expr))
my_user_functions = {"cos": "my_fast_cos"}
custom_ccode = CustomCodePrinter(settings={'user_functions': my_user_functions}).doprint
print(custom_ccode(sp.Rational(1,7) + sp.cos(sp.Rational(1,3))))
,
为了能够在 user_functions
中使用 ccode
,这样做是否是好的做法?
from sympy.printing.c import C99CodePrinter
class CustomCodePrinter(C99CodePrinter):
def _print_Rational(self,expr):
return str(sp.N(expr,n = 17))
def customccode(expr,assign_to=None,**settings):
return CustomCodePrinter(settings).doprint(expr,assign_to)
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。