diff --git a/ex-7/plot.py b/ex-7/plot.py index 9d4515f..40f1d26 100755 --- a/ex-7/plot.py +++ b/ex-7/plot.py @@ -3,7 +3,12 @@ from pylab import * import sys +def line(x, y, **args): + '''line between two points x,y''' + plot([x[0], y[0]], [x[1], y[1]], **args) + w = loadtxt(sys.stdin, max_rows=2) +v = array([[0, -1], [1, 0]]) @ w cut = float(input()) n, m, d = map(int, input().split()) @@ -11,10 +16,13 @@ data = loadtxt(sys.stdin).reshape(n + m, d) signal, noise = data[:n].T, data[n:].T figure() +subplot(aspect='equal') scatter(*signal, c='xkcd:grey blue', label='signal') scatter(*noise, c='xkcd:baby blue', label='noise') -scatter(*(cut * w/np.dot(w,w)), c='red') -arrow(*(-abs(w)), *4*abs(w)) +line(-20*w, 20*w, c='xkcd:blue', label='projection') +line(w-10*v, w+10*v, c='xkcd:red', label='cut') +xlim(-1.5, 8) +ylim(-1.5, 8) legend() figure()