[Numpy-discussion] broadcasting question
Neal Becker
ndbecker2 at gmail.com
Thu Aug 30 07:49:55 EDT 2012
I think this should be simple, but I'm drawing a blank
I have 2 2d matrixes
Matrix A has indexes (i, symbol)
Matrix B has indexes (state, symbol)
I combined them into a 3d matrix:
C = A[:,newaxis,:] + B[newaxis,:,:]
where C has indexes (i, state, symbol)
That works fine.
Now suppose I want to omit B (for debug), like:
C = A[:,newaxis,:]
In other words, all I want is to add a dimension into A and force it to
broadcast along that axis. How do I do that?
More information about the NumPy-Discussion
mailing list