代码之家 › 专栏 › 技术社区 › Isky Mathews

Python-Numba多项式根下误差

numba sympy python

Isky Mathews · 技术社区 · 6 年前

我创建了一个函数,给定系数的范围,用这些系数构造多项式,并输出其所有根的列表。然而,Numba不喜欢它。是这样的:

import math
import numpy as np
import itertools
from numba import jit
from sympy.solvers import solve
from sympy import Symbol
from sympy import Poly

@jit
def polyn(ranges=[[-20,20],[-20,20],[-20,20],[-20,20]],step=4):
    l = []
    x = Symbol('x')
    rangl = [np.linspace(i[0],i[1],math.floor((i[1]-i[0])/step)) for i in ranges]
    coeffl = iter(itertools.product(*rangl))
    leng = 1
    for i in rangl:
        leng *= len(i)
    for i in range(0, leng):
        a = solve(Poly(list(next(coeffl)),x),x)
        for j in a:
            l.append(j)
    return np.array(l)

当我尝试运行此程序时,它会输出一个神秘的: AssertionError:在对象处失败(对象模式前端) 我不明白。。。有人能帮忙吗?

1 回复 | 直到 6 年前

John Zwinck 6 年前

你的代码中有很多东西是Numba目前无法处理的。第一个是你构建的列表理解 rangl :

[np.linspace(i[0],i[1],math.floor((i[1]-i[0])/step)) for i in ranges]

您应该将其替换为NumPy解决方案,如:

rangl = np.empty((len(ranges), step))
for i in ranges:
    rangl[i] = np.linspace(i[0],i[1],math.floor((i[1]-i[0])/step))

第二件Numba无法应付的事情是itertools。产品您也可以用NumPy和for循环来替换它。

一般来说,试着通过注释代码的较低部分来减少代码,直到Numba接受它,然后自上而下地工作,看看哪些部分无法编译。要有条不紊,一步一步走,尽量坚持简单的结构,比如简单 for 循环和数组。

推荐文章

Adam · 绘制不同值的总和

2 年前

Adam · Python中的极限和求和

2 年前

Yoav Tarazy · 拉格朗日乘数/w

2 年前

Layman Tucker · 将辛表达式简化为变量的隐函数

6 年前

mike · 非交换函数乘积的导数

6 年前

dbrane · 将Symphy表达式分解为只涉及一个符号的组

6 年前

Pro Q Rich Lysakowski PhD · 如何在不简化的情况下将字符串转换为Sympy表达式?

6 年前

FLCL · 如何用相应方程式的左侧部分替换表达式的一部分?

6 年前

George · 隐藏泰勒级数的数量级

6 年前

sun0727 · 如何在Python中求解符号方程组?

6 年前