[Numpy-discussion] nicest way to apply an arbitrary sequence of row deltas to an array
Mon Dec 21 03:35:08 CST 2009
George Dahl <george.dahl <at> gmail.com> writes:
> Hi everyone,
> I was wondering if anyone had insight on the best way to solve the
> following problem.
> Suppose I have a numpy array called U.
> U has shape (N,M)
> Suppose further that I have another array called dU and that
> dU has shape (P,M) and that P has no particular relationship to N, it
> could be larger or smaller.
> Additionally, I have a one dimensional array called idx and
> idx has shape (P,). Furthermore idx holds integers that are all valid
> indices into the rows of U. idx almost certainly contains some
> I want,
> for k in range(P):
> U[idx[k],:] += dU[k,:]
> Is there a nice vectorized and efficient way to do this without making
> the obvious python for loop I included above? For what I am doing, P
> is usually quite large. I am most interested in a clever use of
> numpy/scipy commands and Python and not Cython.
I'm also interested to see if there are any answers to this; I came across a
similar problem recently. It would have been convenient to do something like
U[idx] += dU, but this didn't work because there were repeated indices in idx.
Here's a short example that shows the problem:
In : U = np.array([1., 2., 3., 4.])
In : dU = np.array([0.1, 0.1, 0.1, 0.1])
In : idx = np.array([0, 1, 2, 0])
In : U
Out: array([ 1., 2., 3., 4.])
In : U[idx]
Out: array([ 1., 2., 3., 1.])
In : U[idx] += dU
In : U
Out: array([ 1.1, 2.1, 3.1, 4. ])
Ideally U would end up as array([ 1.2, 2.1, 3.1, 4. ])
More information about the NumPy-Discussion