Subclassing numpy.ndarray

Question

I want to implement a subclass of numpy.ndarray that overrides the constructor with something like this:

class mymat(numpy.ndarray):
    def __new__(cls, n, ...):
        ret = np.eye(n)

        # make some modifications to ret

        return ret

Unfortunately, the type of the object returned by this constructor is not cls, but rather numpy.ndarray.

Setting the class of ret with

ret.__class__ = cls # bombs: TypeError: __class__ assignment: only for heap types

won't work.

One possible solution would be something like

class mymat(numpy.ndarray):
    def __new__(cls, n, ...):
        ret = super(mymat, cls).__new__(cls, (n, n))
        ret[:] = np.eye(n)

        # make some modifications to ret

        return ret

This is fine for small n, but I'd prefer to avoid the extra Python-side assignment when n is large.

Is there some other approach that would avoid this extra assignment, and still produce an object of class mymat?

Just making sure you're aware of docs.scipy.org/doc/numpy/user/basics.subclassing.html — NPE
– NPE, Commented Oct 9, 2014 at 18:19
I think you want view casting which should be explained in NPE's link. — mgilson
– mgilson, Commented Oct 9, 2014 at 18:20

elyase · Accepted Answer · 2014-10-09 18:32:48Z

4

Try this:

class mymat(np.ndarray):
    def __new__(cls, n):
        ret = np.eye(n)
        return ret.view(cls)

answered Oct 9, 2014 at 18:32

elyase

41.2k12 gold badges121 silver badges123 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Subclassing numpy.ndarray

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related