ludlcs91

Como coloco cor no código abaixo, preciso plota cada resultado de uma cor, alguém pode me ajudar import numpy as np import csv import matplotlib.pyplot as plt def open_file(path): with open(path) as dataset: data = np.array(list(csv.reader(dataset))) labels = np.array(list(set(data[1:,-1]))) header = data[0] x_data = np.zeros((len(data)-1,len(data[0])-1)) y_data = np.empty(len(data)-1) for x in range(1,len(data)): x_data[x-1] = data[x][:-1] for y in range(len(labels)): if labels[y] in data[x]: y_data[x-1] = y return header,x_data,y_data header,x_data,y_data = open_file("iris_2classes.csv") plt.scatter(x_data[:,1],x_data[:,0],c=y_data) plt.title("Setosa x versicolor" ) plt.xlabel('Petal.Width') plt.ylabel('Petal.Length') plt.show()