[Numpy-discussion] np.choose() question

Andreas Hilboll lists at hilboll.de
Tue Jun 8 12:24:51 EDT 2010


Hi there,

I have a problem, which I'm sure can somehow be solved using np.choose() 
- but I cannot figure out how :(

I have an array idx, which holds int values and has a 2d shape. All 
values inside idx are 0 <= idx < n. And I have a second array times, 
which is 1d, with times.shape = (n,).

Out of these two arrays I now want to create a 2d array having the same 
shape as idx, and holding the values contained in times, as indexed by
idx.

A simple np.choose(idx,times) does not work (error "Need between 2 and 
(32) array objects (inclusive).").

Example:

idx = [[4,2],[3,1]]
times = [100,101,102,103,104]

 From these two I want to create an array

result = [[104,102],[103,101]]

How can this be done?

Thanks a lot for your insight!

Andreas




More information about the NumPy-Discussion mailing list