gistfile1.py

import matplotlib.pyplot as plt

dataset = open("assignment1_train.txt", "rb")

x = []
y = []
color=[]

for line in dataset:
	line = line.strip()
	line = line.split(" ")
	print(line)
	x.append(line[0])
	y.append(line[1])
	if line[2] == "1":
		color.append("red")
	else:
		color.append("green")

plt.scatter(x, y, s=25, c=color, alpha=1)
plt.show()