Stefan van der Walt wrote:
> Hi Christiaan
> On Sun, Apr 15, 2007 at 02:03:49PM +0900, Christian K wrote:
>> could someone please provide example code for how to make a subclassed 
>> ndarray
>> pickable? I don't quite understand the docs of ndarray.__reduce__.
>> My subclassed ndarray has just one additional attribute.
> __reduce__ is part of the pickle protocol, which is described at
> You need to modify __reduce__ to store the complete state of your
> custom object, as well as __setstate__ which restores the state on
> unpickling.
> See the attached code as an example.  There, I have an InfoArray,
> which is a subclassed numpy.ndarray.  The InfoArray contains an
> additional member, info, which is a dictionary.
> The __reduce__ method calls ndarray's __reduce__, then adds to the
> result the state of the InfoArray.  Conversely, the __setstate__
> method calls ndarray's __setstate__ as well as restoring the info
> member.

Thank you for the explanation and the example. Works very well.


Numpy-discussion mailing list

Reply via email to