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] 中的元组列表执行此操作。

希望得到一些帮助！

亲切的问候

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）