python - Shading background based on groups above/below a line -


suppose have scatterplot kind of line (least squares regression line, knn regression line, etc.) through it, this. enter image description here want shade upper region of plot reddish, , lower region of plot blueish, give indication of how line doing classifier points. similar mimic example effect plot elements of statistical learning (hastie et al), (chapter 2, page 13).

enter image description here

how can achieve effect matplotlib?

i know how set rectangular regions of plot different colors axhspan , axvspan (see this answer), have been struggling set different plot colors based on regions above , below line.

code replicate current mock plot

import numpy np import matplotlib.pyplot plt  plt.style.use('seaborn-notebook')  np.random.seed(17) grp1_x = np.random.normal(1, 1, 100) grp1_y = np.random.normal(3, 1, 100)  grp2_x = np.random.normal(1.2, 1, 100) grp2_y = np.random.normal(1.2, 1, 100)  ######################################## ## least squares plot  plt.scatter(grp1_x, grp1_y,             lw         = 1,             facecolors = 'none',             edgecolors = 'firebrick') plt.scatter(grp2_x, grp2_y,             lw         = 1,             facecolors = 'none',             edgecolors = 'steelblue') plt.tick_params(     axis        = 'both',           = 'both',     bottom      = 'off',     top         = 'off',     labelbottom = 'off',     right       = 'off',     left        = 'off',     labelleft   = 'off')  full_x = np.concatenate([grp1_x, grp2_x]) full_y = np.concatenate([grp1_y, grp2_y]) m, c = np.linalg.lstsq(np.vstack([full_x,                                   np.ones(full_x.size)]).t,                        full_y)[0] plt.plot(full_x, m*full_x + c, color='black') plt.show()

first recommend sorting x values, such line looks smooth.

x = np.sort(full_x) plt.plot(x, m*x + c, color='black')

then can use fill_between fill region above (below) line (from) upper (lower) plot limits.

xlim=np.array(plt.gca().get_xlim()) ylim=np.array(plt.gca().get_ylim()) plt.fill_between(xlim, y1=m*xlim + c, y2=[ylim[0],ylim[0]],                   color="#e0eaf3", zorder=0 ) plt.fill_between(xlim, y1=m*xlim + c, y2=[ylim[1],ylim[1]],                   color="#fae4e4", zorder=0 ) plt.margins(0)

enter image description here

or use hatching background:

fb1 = plt.fill_between(xlim, y1=m*xlim + c, y2=[ylim[0],ylim[0]],                   facecolor="w", edgecolor="#e0eaf3", zorder=0 ) fb1.set_hatch("//") fb2 = plt.fill_between(xlim, y1=m*xlim + c, y2=[ylim[1],ylim[1]],                    facecolor="w", edgecolor="#fae4e4", zorder=0 ) fb2.set_hatch("\\\\")

enter image description here


Comments

Post a Comment

Popular posts from this blog

python - Operations inside variables -

this question has answer here: evaluating mathematical expression in string 12 answers i want make python script finds maximum possible output using 2 operations. problem says there syntax error @ 2nd last line, in b in there. how can fix it? x = 0 = int(input("1st number:")) c = int(input("2nd number:" )) e = int(input("3rd number:" )) in range(4): if == 0: b = "+" elif == 1: b = "-" elif == 2: b = "/" else: b = "*" j in range(4): if j == 0: d = "+" elif j == 1: d = "-" elif j == 2: d = "/" else: d = "*" k = b c d e print(k) if you're after variables e on single line, couldn...
Read more

Generic Map Parameter java -

i having lot of methods of type: public static list<eduusers> getdetails(class c, map<string, integer> params) { list<eduusers> details = null; session session = hibernateutil.getsessionfactory().opensession(); transaction tx = null; try { tx = session.begintransaction(); criteria cr = session.createcriteria(c); (map.entry<string, integer> entry : params.entryset()) { cr.add(restrictions.eq(entry.getkey(), entry.getvalue())); } details = cr.list(); tx.commit(); } catch (exception asd) { log.debug(asd.getmessage()); if (tx != null) { tx.commit(); } } { session.close(); } return details; } i trying have generic method them , have written far: public static <t> list<t> getdetails(class<t> c, class<t> b) { l...
Read more

arrays - What causes a java.lang.ArrayIndexOutOfBoundsException and how do I prevent it? -

what arrayindexoutofboundsexception mean , how rid of it? here code sample triggers exception: string[] name = {"tom", "dick", "harry"}; for(int = 0; i<=name.length; i++) { system.out.print(name[i] +'\n'); } your first port of call should documentation explains reasonably clearly: thrown indicate array has been accessed illegal index. index either negative or greater or equal size of array. so example: int[] array = new int[5]; int boom = array[10]; // throws exception as how avoid it... um, don't that. careful array indexes. one problem people run thinking arrays 1-indexed, e.g. int[] array = new int[5]; // ... populate array here ... (int index = 1; index <= array.length; index++) { system.out.println(array[index]); } that miss out first element (index 0) , throw exception when index 5. valid indexes here 0-4 inclusive. correct, idiomatic for statement here be: for (int index = 0; index <...
Read more