c = ["blue" if i==0 else "red" if i==1 else "green" for i in y]