[Numpy-discussion] Help w/ indexing, please

2010-07-27 Thread David Goldsmith
Hi! I have a large M x K, M, K ~ 1e3 array L of indices - non-negative integers in the range 0 to N-1 - and an N x 3 array C (a matplotlib colormap). I need to create an M x K x 3 array R such that R[m,k,j] = C[L[m,k], j], j = 0,1,2. I want to do so w/out having to loop through all the (m,k)

Re: [Numpy-discussion] Help w/ indexing, please

2010-07-27 Thread John Salvatier
I am pretty sure you should be able to do R = C[L, :] and get the array you want. Try it with a small matrix where you know the result you want. You may need to transpose some axes afterwards, but I don't think you should. On Tue, Jul 27, 2010 at 9:10 AM, David Goldsmith

Re: [Numpy-discussion] Help w/ indexing, please

2010-07-27 Thread David Goldsmith
On Tue, Jul 27, 2010 at 9:32 AM, John Salvatier jsalv...@u.washington.eduwrote: I am pretty sure you should be able to do R = C[L, :] and get the array you want. Try it with a small matrix where you know the result you want. You may need to transpose some axes afterwards, but I don't