Как построить 2D-точки разных цветов на основе метки y в Python?
Я пытаюсь визуализировать вывод моего алгоритма кластеризации на наборе данных Iris с разбросанным графиком. Мой список X содержит список точек (4D), а Y содержит новую метку, созданную алгоритмом.
Например,
Если X= =[ [1, 2, 3, 4], [4, 5, 6, 7], [9, 10, 11, 15], [1, 6, 7, 3], [7, 4, 2, 1] ]
и Y= [2, 1, 0, 2, 0]
Я хочу изобразить это на разбросанном графике на основе метки Y. Например, точки метки 0 могут быть окрашены в красный цвет, 1 - в желтый цвет и 2 - в синий цвет. Я знаю, что невозможно построить 4D-точки, но мы выбираем пару (x1,x2) и строим ее с цветами, определенными их меткой y.
Как сделать то же самое?