[Numpy-discussion] arbitrary number of newaxis
Andrew Corrigan
acorriga at gmu.edu
Mon Jul 3 08:36:53 EDT 2006
In a function I'm writing, I multiply two arrays together: A and B.
where A.ndim == 2 and I don't know B.ndim in advance
If I knew B.ndim == 3, then I would write
A[:,:,newaxis,newaxis,newaxis]*B[newaxis,newaxis,...]
or if I knew that B.ndim == 1 then I would write
A[:,:,newaxis*B[newaxis,newaxis,...]
but I don't know B.ndim. Essentially I want to say something like:
A[:,:,repeat(newaxis, B.ndim)]*B[newaxis,newaxis,...]
How can I express what I mean, such that it actually works?
More information about the NumPy-Discussion
mailing list