diff --git a/ex-7/plot.py b/ex-7/plot.py index 40f1d26..9627766 100755 --- a/ex-7/plot.py +++ b/ex-7/plot.py @@ -7,6 +7,8 @@ def line(x, y, **args): '''line between two points x,y''' plot([x[0], y[0]], [x[1], y[1]], **args) +plt.rcParams['font.size'] = 30 + w = loadtxt(sys.stdin, max_rows=2) v = array([[0, -1], [1, 0]]) @ w cut = float(input()) @@ -17,18 +19,17 @@ 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(*signal, edgecolor='#51492d', c='#dbbb4a', label='signal') +scatter(*noise, edgecolor='#512d3d', c='#d64d88', label='noise') 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() sig_proj = np.dot(w, signal) noise_proj = np.dot(w, noise) -hist(sig_proj, color='xkcd:grey blue') -hist(noise_proj, color='xkcd:baby blue') +hist(sig_proj, color='#dbbb4a') +hist(noise_proj, color='#d64d88') axvline(cut, c='r') show()