DecisionBoundryColor

gistfile1.txt

#%matplotlib notebook
%matplotlib inline
import numpy as np
import matplotlib.pyplot as plt
from matplotlib import colors as c

import time

fig = plt.figure()
ax = fig.gca()
resolution=200
xs = np.linspace(-3., 3., resolution)
ys = np.linspace(-3., 3., resolution)
xx, yy = np.meshgrid(xs, ys)
zz=2*xx+4*yy-5
zz[zz<0]=-1
zz[zz>0]=+1

cMap = c.ListedColormap(['r','g'])

quad = ax.pcolormesh(xs, ys, zz,cmap=cMap)
positive = np.array([[2, 2], [1, 2]])
negative = np.array([[0, 0], [-1, -2]])
plt.plot(positive[:, 0], positive[:,1], 'ro', markersize=10)
plt.plot(negative[:, 0], negative[:,1], 'yo', markersize=10)
plt.show()