[Numpy-discussion] return type from ufuncs

Marek Wojciechowski mwojc at p.lodz.pl
Thu Nov 20 12:47:41 EST 2014


Hi!

I wrote a simple subclass of np.ndarray and now i do call np.sum() on it. I 
expected that the result will be a python float (or int) just like when summing 
up regular arrays. Instead i obtain the (scalar) view of my subclass. How can 
i change this behavior? I tried writing __array_wrap__ method like this: 

def __array_wrap__(self, out_arr, context=None):
	selfv = self.view(np.ndarray)
        return np.ndarray.__array_wrap__(selfv, out_arr, context)

but this just returns np.ndarray type and not float.

Regards,
-- 
Marek Wojciechowski 




More information about the NumPy-Discussion mailing list