diff --git a/pykat/utilities/plotting/tools.py b/pykat/utilities/plotting/tools.py index 60599c64507c8ff90d90a770a38f8c9efaecd614..35e46f25f43e1778f2ce531723973db64a78e2c2 100644 --- a/pykat/utilities/plotting/tools.py +++ b/pykat/utilities/plotting/tools.py @@ -18,6 +18,10 @@ formatter = matplotlib.ticker.EngFormatter(unit='', places=0) formatter.ENG_PREFIXES[-6] = 'u' +def argmaxM(data): + return(np.unravel_index(np.argmax(data),data.shape)) + + def printPDF(fig, filename): if filename !='' and filename != None: if not filename.lower().endswith(('.pdf')): @@ -29,11 +33,12 @@ def printPDF(fig, filename): def plot_power_contour(x,y,z,xlabel, ylabel, clabel, title='', filename=''): ax,fig=plot_setup() xm,ym=np.meshgrid(x,y) - extent = [x[0], x[-1], y[0], y[-1] ] + extent = [x[0], x[-1], y[0], y[-1]] mycm='YlOrRd_r' - mycm='hot' + #mycm='hot' #im = ax.imshow(z,origin='lower',extent=extent,cmap=mycm, aspect='auto', interpolation='nearest') im = ax.imshow(z,origin='lower',extent=extent,cmap=mycm, aspect='auto') + #im = ax.imshow(z,origin='lower',cmap=mycm, aspect='auto') cb = fig.colorbar(im, format="%.4g") #cb.set_clim(-1.0*zlimit, zlimit) ax.autoscale(False) @@ -45,10 +50,12 @@ def plot_power_contour(x,y,z,xlabel, ylabel, clabel, title='', filename=''): ax.xaxis.set_major_formatter(formatter) ax.yaxis.set_major_formatter(formatter) fig.canvas.manager.set_window_title(title) - plt.show() + plt.show(block=0) printPDF(fig, filename) + + def plot_error_contour(x,y,z,xlabel, ylabel, clabel, title='', filename=''): global fig, ax, cb, ct, data rc('font',**pp.font)