[Numpy-discussion] Multiply along axis
Robert Elsner
mlist at re-factory.de
Wed Jun 29 10:32:14 EDT 2011
Hello everyone,
I would like to solve the following problem (preferably without
reshaping / flipping the array a).
Assume I have a vector v of length x and an n-dimensional array a where
one dimension has length x as well. Now I would like to multiply the
vector v along a given axis of a.
Some example code
a = np.random.random((2,3))
x = np.zeros(2)
a * x # Fails because not broadcastable
So how do I multiply x with the columns of a so that for each column j
a[:,j] = a[:,j] * x
without using a loop. Is there some (fast) fast way to accomplish that
with numpy/scipy?
Thanks for your help
Robert Elsner
More information about the NumPy-Discussion
mailing list