Как заставить функцию 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 МПФ. Итак, какой самый быстрый/лучший способ взять функцию в mpmath и преобразовать ее в функцию, которая может действовать на массивы numpy, но это возвращает стандартные float, а не mpmath float? Или просто преобразовать массив mpf в numpy float? функция float() работает для преобразования чисел, но не массивов.

Answer 1

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

должен это сделать. Расчет mpmath, вероятно, будет намного медленнее, чем лямбда, поэтому влияние скорости будет, по-видимому, примерно равным конверсии поплавка.