Имея массив (2d) с индексом цветов, есть ли способ заменить индекс его цветом за один шаг, а не поэлементно?
Пример:
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]]
я пробовал colors[im_x] но это не работает :(
Не могли бы вы быть немного более конкретным?






Определите массивы как массивы numpy, а затем проиндексируйте их, как вы пытались:
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]]])
Проблема в том, что вы не определили массивы...