Векторизация поиска корня в numpyPython

Программы на Python
Anonymous
Векторизация поиска корня в numpy

Сообщение Anonymous »

Мне нужно запустить функцию, которая находит корень уравнения, зависящего от одного параметра, на большом наборе данных (реальное уравнение гораздо сложнее, но формально эквивалентно).

Код: Выделить всё

def f(x):
return numpy.optimize.brentq(lambda y:numpy.exp(-abs(x)*y)-y,0,1)
Удобно ли его векторизовать?
Я пытался использовать numpy.vectorize(f), он работает нормально, но я думаю с той же скоростью, что и цикл Python for.

Есть ли какая-то ручная векторизация, чтобы лучше использовать возможности вычислений массива numpy?>

Вернуться в «Python»