Array of Index to colors to array of colors in python [duplicate]

Question

Having an array (2d) with index of colors is there a way to replace the index by its color in one step not element by element?

Example:

colors=[[255,0,0],[125,222,11]]
im_x=[[0,0],[1,0]]

#the result must be:

im_c= [[[255,0,0],[255,0,0]],[[125,222,11],[255,0,0]]

I have try colors[im_x] but it does not work :(

The problem is that you have not defined arrays...

yatu
– yatu

2019-05-01 16:34:39 +00:00
Commented May 1, 2019 at 16:34 — yatu
– yatu, Commented May 1, 2019 at 16:34
Can you please be a bit more specific?

Jacob
– Jacob

2019-05-01 16:35:02 +00:00
Commented May 1, 2019 at 16:35 — Jacob
– Jacob, Commented May 1, 2019 at 16:35

yatu · Accepted Answer · 2019-05-01 16:35:35Z

1

Define the arrays as numpy arrays, and then index them as you were trying to:

import numpy as np
colors = np.array([[255,0,0],[125,222,11]])
im_x = np.array([[0,0],[1,0]])

colors[im_x]

array([[[255,   0,   0],
        [255,   0,   0]],

       [[125, 222,  11],
        [255,   0,   0]]])

answered May 1, 2019 at 16:35

yatu

88.6k12 gold badges93 silver badges148 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Array of Index to colors to array of colors in python [duplicate]

1 Answer 1

Comments

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Linked

Related