我刚刚开始使用numexpr,尽管github存储库似乎有一些使用它的基本示例,但我不清楚如何将其应用于某些复杂情况。假设我有一个函数:

def func(x):
  #x is a numpy array of very large size
  return 0.5*np.exp(-x**2)*x**(1/3)+np.exp(-x)*(1.5*np.sqrt(x)+0.3/(1.+x))


numexpr编写此代码的等效方式是什么?

最佳答案

np.sqrtsqrt; np.expexp

import numexpr as ne
y = ne.evaluate(
    '.5 * exp(-x ** 2) * x ** (1 / 3)'
    '+ exp(-x) * (1.5 * sqrt(x)'
    '+ .3 / (1. + x))'
)

关于python - 使用numexpr优化numpy中的数值运算,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/50298590/

10-09 15:15