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