Robert Kern wrote: > On Mon, Jun 21, 2010 at 14:01, Neal Becker <[email protected]> wrote: >> Can I find an efficient way to do this? >> >> I have a 2d array, A, 80 rows by 880 columns. >> >> I have a vector, B, of length 80, with scalar indexes. > > I assume you mean 880. > >> I want a vector output C where >> C[i] = A[b[i],i] (i=0,879) > > C = A[b, np.arange(880)] >
Thanks! Just what I needed. I wouldn't have guessed this. Do we have a wiki to save useful examples like this? _______________________________________________ NumPy-Discussion mailing list [email protected] http://mail.scipy.org/mailman/listinfo/numpy-discussion
