def test1(): fig,ax = subplots(nrows=3, ncols=2, apportion = [2,1,1]) return
bb = ax.get_position().get_points() pos = bb.flatten() # convert to x0,y0, dx, dy form by subtracting origin newh = height[i]*fracts[i]*nrows pos[1] = yabove - newh - above[i] pos[3] = newh pos[2] = pos[2] - pos[0] yabove = pos[1] if debug>0: print(pos) ax.set_position(pos) if squeeze: if len(np.shape(axes[0]))==0: axes = axes.flatten() if len(axes) == 1: axes = axes[0] return(fig, axes) if __name__ == "__main__": from subplots import subplots fig,ax = subplots(nrows=3, apportion = [5,1,1]) #fig,ax = pl.subplots(nrows=3) t = np.linspace(0,20) ax[0].plot(np.sin(t)) ax[0].set_title('Ax0') ax[1].plot(np.cos(t)) pl.show() print('try test1()') def test1(): fig,ax = subplots(nrows=3, ncols=2, apportion = [2,1,1]) return
exec(process_cmd_line_args()) pf.config.set('global', 'localdatapath', os.path.join(pf.PYFUSION_ROOT_DIR, 'examples', 'JSPF_tutorial', 'local_data')) # this line points to local data if nr > 0: try: # use my nicer version if it is available from subplots import subplots # close up space between graphs spkwargs = dict(apportion=[2, 1], spadj_kwargs=dict(hspace=0)) except ImportError: from matplotlib.pyplot import subplots spkwargs = {} fig, axs = subplots(nrows=nr, ncols=1, sharex='all', **spkwargs) if nr == 1: strtpair = 0 ax1, ax2 = axs, axs plt.sca(ax2) else: strtpair = 1 ax1, ax2 = axs plt.sca(axs[0]) xlab = {1: 'Hz', 1000: 'kHz', 1000000: 'MHz'}[unit_freq] if type(noverlap) != int: noverlap = int(noverlap * NFFT)
khstr = '$\kappa_H$ {kh:.2f}'.format(kh=kh) try: rf_drive=h1tree.getNode('.RF:rf_drive').data() except: PFTop = h1tree.getNode('.RF:P_FWD_TOP').data() rf_drive = 1.3*5*np.sqrt(np.max(smooth_n(PFTop,500)/200.)) if include_amplifier_sig: current_node = tree_name.getNode(mirnov) current_raw = current_node.record.data() nr = 1+include_ne+include_ne_spec+include_amplifier_sig (fig, ax) = subplots(nrows = nr, sharex='all', squeeze=False, **special_kws) rw = 0 """ rw = 1; nr += 1 # leave two spots for the specgram pos = msp.get_position().get_points().flatten() print(pos) msp.set_position([pos[0],pos[1]*2./nr,pos[2],pos[3]*nr/(nr-1.)]) """ msp = ax[rw,0] if include_amplifier_sig: rw += 1 asp = ax[rw, 0] if include_ne:
from pyfusion.utils import process_cmd_line_args exec(process_cmd_line_args()) pf.config.set('global','localdatapath',os.path.join(pf.PYFUSION_ROOT_DIR,'examples','JSPF_tutorial','local_data')) # this line points to local data if nr > 0: try: # use my nicer version if it is available from subplots import subplots # close up space between graphs spkwargs=dict(apportion=[2,1],spadj_kwargs=dict(hspace=0)) except ImportError: from matplotlib.pyplot import subplots spkwargs={} fig, axs = subplots(nrows=nr,ncols=1, sharex='all',**spkwargs) if nr==1: strtpair = 0 ax1, ax2 = axs, axs plt.sca(ax2) else: strtpair = 1 ax1, ax2 = axs plt.sca(axs[0]) xlab = {1:'Hz', 1000:'kHz', 1000000:'MHz'}[unit_freq] if type(noverlap) != int: noverlap = int(noverlap*NFFT)