Dúvida exercicio alguém me ajuda

[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()

[slackware15]

Eu aprendi lendo a wiki do python .. se pesquisar acha, mas já que esta aqui:

 

>>> print '\033[31m'+'Isto é vermelho'+'\033[0;0m'
   2 Isto é vermelho
   3 >>> print '\033[32m'+'Isto é verde'+'\033[0;0m'
   4 Isto é verde
   5 >>> print '\033[42m'+'\033[1m'+'\033[33m'+'Isto é amarelo negrito com fundo verde'+'\033[0;0m'
   6 Isto é amarelo negrito com fundo verde

 

 

vermelho = '\033[31m'
   2 verde = '\033[32m'
   3 azul = '\033[34m'
   4 
   5 ciano = '\033[36m'
   6 magenta = '\033[35m'
   7 amarelo = '\033[33m'
   8 preto = '\033[30m'
   9 
  10 branco = '\033[37m'
  11 
  12 restaura cor original = '\033[0;0m'
  13 negrito = '\033[1m'
  14 reverso = '\033[2m'
  15 
  16 fundo preto = '\033[40m'
  17 fundo vermelho = '\033[41m'
  18 fundo verde = '\033[42m'
  19 fundo amarelo = '\033[43m'
  20 fundo azul = '\033[44m'
  21 fundo magenta = '\033[45m'
  22 fundo ciano = '\033[46m'
  23 fundo branco = '\033[47m'