[Numpy-discussion] custom accumlators
tim.hochberg at ieee.org
Fri Jan 5 17:51:14 EST 2007
A. M. Archibald wrote:
> Really it would be nice if what vectorize() returned were effectively
> a ufunc, supporting all the various operations we might want from a
> ufunc (albeit inefficiently). This should not be difficult, but I am
> not up to writing it this evening.
You might want to look at frompyfunc:
def expmave2(x, k):
def expmave_sub(a, b):
return a + k * (b - a)
return np.frompyfunc(expmave_sub, 2, 1).accumulate(x)
It's amazing what you find when you dig around.
More information about the NumPy-Discussion