Python：比较NumPy和SymPy-ufuncified函数的速度

Question

刚刚写了一个代码来比较一个函数的计算速度，这个函数写为numpy，另一个函数使用ufuncifyfrom{}：

import numpy as np
from sympy import symbols, Matrix
from sympy.utilities.autowrap import ufuncify
u,v,e,a1,a0 = symbols('u v e a1 a0')

dudt = u-u**3-v 
dvdt = e*(u-a1*v-a0)

p = {'a1':0.5,'a0':1.5,'e':0.1}

eqs = Matrix([dudt,dvdt])

numeqs=eqs.subs([(a1,p['a1']),(a0,p['a0']),(e,p['e'])])
print eqs
print numeqs

dudt = ufuncify([u,v],numeqs[0])
dvdt = ufuncify([u,v],numeqs[1])

def syrhs(u,v):
    return dudt(u,v),dvdt(u,v)

def nprhs(u,v,p):
    dudt = u-u**3-v 
    dvdt = p['e']*(u-p['a1']*v-p['a0'])
    return dudt,dvdt

def compare(n=10000):
    import time
    timer_np=0
    timer_sy=0
    error = np.zeros(n)
    for i in range(n):
        u=np.random.random((128,128))
        v=np.random.random((128,128))
        start_time=time.time()
        npcalc=np.ravel(nprhs(u,v,p))
        mid_time=time.time()
        sycalc=np.ravel(syrhs(u,v))
        end_time=time.time()
        timer_np+=(mid_time-start_time)
        timer_sy+=(end_time-mid_time)
        error[i]=np.max(np.abs(npcalc-sycalc))
    print "Max difference is ",np.max(error), ", and mean difference is ",np.mean(error)
    print "Average speed for numpy ", timer_np/float(n)
    print "Average speed for sympy ", timer_sy/float(n)

在我的机器上，结果是：

^{pr2}$

任何关于如何使上述功能更快的建议都是欢迎的！在

Answer 1

经过进一步的探索，似乎ufuncify和正则numpy函数的计算速度大致相同。使用numba或打印到theano函数并没有产生更快的代码。因此，另一个加快速度的方法是cython或包装c或{}代码。在