从阵列创建mpf

Question

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

下面是一个重现错误的最小示例。对于这个例子，从技术上讲，我不需要mpmath包，但是我实际的函数包含了可以实现的超Confluent函数。在

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

错误的具体原因是：

(Pdb) x0
array([mpf('1.0')], dtype=object)
(Pdb) mp.sin(x0)
*** TypeError: cannot create mpf from array([mpf('1.0')], dtype=object)

结果是，fsolve试图将您的estimate转换为数组，而numpy不知道如何处理mpmath对象。在

^{pr2}$

改变fsolve的工作方式并不是很有成效，所以最好的办法似乎是教您的函数处理mpmath对象数组

def cos_func(p):
   vn = p
   if isinstance(p, np.ndarray):
         if p.size == 0: 
             vn = p[0]
         else:
             raise ValueError  # or whatever you want to do here"
   return mp.sin(vn)

Answer 2

诀窍是将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