python - 从数组创建 mpf

Question

我正在尝试将 fsolve 与 mpmath 包结合使用。但是，我收到错误 cannot create mpf from array([mpf('1.0')], dtype=object)。

这是一个重现错误的最小示例。对于此示例，从技术上讲，我不需要 mpmath 包，但我的实际函数包含需要的超融合函数。

from scipy.optimize import fsolve
#from mpmath import hyp1f1 as hyp1f1mp
#from mpmath import gamma as gammamp
import mpmath as mp
#import numpy as np

mp.dps = 250; mp.pretty = True


def cosFunc(p):
   vn = p
   output = mp.sin(vn)
   return output

estimate = mp.mpf(1)
value = fsolve(cosFunc,estimate)
print value

我发现一个类似的问题建议使用 np.frompyfunc ( How to mpf an array? )，但它告诉我该函数不可调用(当我将其应用于 vn 时) >).

Answer 1

技巧是将 np.frompyfunc 应用于函数而不是值。我认为以下修改将使您的功能正常工作:

def cosFunc(p):
  vn = p
  np_sin = np.frompyfunc(mp.sin, 1, 1)
  output = np_sin(vn)
  return float(output)

value = fsolve(cosFunc, 1)
print value