multiply numpy ndarray with 1d array along a given axis

Question

It seems I am getting lost in something potentially silly. I have an n-dimensional numpy array, and I want to multiply it with a vector (1d array) along some dimension (whi

Answer 1

You could also use a simple matrices trick

c = np.matmul(a,diag(b))

basically just doing matrix multiplication between a and a matrix whose diagonals are the elements of b. Maybe not as efficient but it's a nice single line solution