Как построить pcolor-изображение для разных calsses, используя matplotlib?

Question

Я знал кое-что о pcolor изображении, используя matplotlib для простого набора данных. однако, если у меня есть два набора данных, включая данные классов и набор атрибутов, связанные с набором данных классов.

classes_dataset = np.array([["a","b","b"],["a","c","a"],["b","c","c"]])
attribute_dataset = np.array([[0.1,0.9,0.6],[0.2,0.4,0.2],[0.7,0.8,1.0]])

, где 0,1,0,2 и 0,2 - атрибут для класса a, 0,9,0,6 и 0,7 - атрибут для класса b, а 0,4,0,8 и 1,0 - атрибут для класса c

Таким образом, я хочу, чтобы pcolor-изображение содержало информацию как о классах, так и об атрибутах. Для класса a, b, c карта будет другой.

искренне спасибо

Answer 1

Преобразуйте массив вашего класса в индекс, затем при построении графика укажите для массива значение C, чтобы задать цвет, и, если v - это вектор с индексом для каждой точки, используйте:

C=attribute_dataset[v]