Python Script

Python script work but not sure how to give colors of my binary class so that I can see 2 classes when plotted in 2D.
Any code that help me to implement in python script?
Thanks