Я новичок в Numpy.Я пытаюсь написать программу визуализации данных, и мне нужно превратить сетку чисел в сетку цветов.Должно быть довольно очевидно, как я пытался это сделать:
import numpy as np
import pdb
colors = np.array(["red","blue","green","yellow"])
data = np.random.choice([0,1,2,3],(4,4))
def color_in_square(val, cmap):
return cmap[val]
color_grid = np.vectorize(color_in_square)
display = color_grid(data,colors)
print(display)
Этот код не работает как есть.Это работает, если я вызываю цвета из color_in_square () как глобальную переменную, но функция vectorize () не позволит мне проходить через массивы разного размера в качестве входных данных, которые мне нужны для работы с остальной частью моего модуля.Чего мне не хватает, чтобы быть в состоянии сделать это, или я все это делаю неправильно?Спасибо