[Numpy-discussion] subclassing matrix
Basilisk96
basilisk96 at gmail.com
Fri Jan 11 23:59:55 EST 2008
On Jan 11, 2008, Colin J. Williams wrote:
> You make a good case that it's good not
> to need to ponder what sort of
> vector you are dealing with.
>
> My guess is that the answer to your
> question is "no" but I would need to
> play with your code to see that. My
> feeling is that, at the bottom of
> the __new__ module, the returned object
> should be an instance of the
> Vector class.
>
> It's been a while since I've worked with
> numpy and so I'll look at it
> and hope that someone gives you a
> definitive answer before I sort it out.
>
> Colin W.
Well, let's say that I get rid of that class promotion line. When the
input object to the constructor is a string or a tuple
such as Vector('1 2 3') or Vector([1,2,3]), then the returned object
is always an instance of Vector. However, when the input object is a
numpy.matrix instance, the returned object remains a numpy.matrix
instance! So by doing that little hack, I promote it to Vector.
BUT...
It seems that I have solved only half of my problem here. The other
half rears its ugly head when I perform operations between instances
of numpy.matrix and Vector. The result ends up returning a matrix,
which is bad because it has no knowledge of any custom Vector
attributes. Here's a simple case:
u = Vector('1 2 3') #Vector instance
P = numpy.mat(numpy.eye(3)) #matrix instance
u_new = P*u #matrix instance, not desirable!
u_new_as_Vector = Vector(P*u) #Vector instance
I'd rather not have to remember to re-instantiate the result in client
code. I think I understand why this is happening - the code in
numpy.core.defmatrix.matrix.__mul__ goes like this:
def __mul__(self, other):
if isinstance(other,(N.ndarray, list, tuple)) :
# This promotes 1-D vectors to row vectors
return N.dot(self, asmatrix(other))
if N.isscalar(other) or not hasattr(other, '__rmul__') :
return N.dot(self, other)
return NotImplemented
It passes the first condition: isinstance(other,(N.ndarray)) is true;
and so the return value becomes a matrix.
Bummer.
Do I also need to override a few overloaded methods like __mul__,
__rmul__, etc. to make this work?
Cheers,
-Basilisk96
More information about the NumPy-Discussion
mailing list