python - 如何使 mpmath 函数返回 python float？

Question

我想将 mpmath 函数转换为可以在 numpy 数组上运行的函数。假设我有以下示例

A=np.linspace(0,1,100)
besseli_vec = numpy.frompyfunc(mpmath.besseli, 2, 1)
Y=besseli_vec(0, A)

但是，现在数组 A 中的值是 mpmath 类型 mpf。那么，在 mpmath 中获取函数并将其转换为可以作用于 numpy 数组但返回标准 float 而不是 mpmath float 的函数的最快/最佳方法是什么？或者只是将 mpf 数组转换为 numpy float ？函数 float() 可转换数字，但不能转换数组。

Answer 1

besseli_vec = numpy.frompyfunc(lambda *a: float(mpmath.besseli(*a)), 2, 1)

应该这样做。 mpmath 计算可能会比 lambda 慢得多，因此速度影响可能大约只是浮点转换。