[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