def plot2d_unstructured_nodes(self, data, *args, **kwargs): method = kwargs.get('plot_function', 'tricontourf') x1 = self.x[:,0] x2 = self.x[:,1] xmin = min(x1) xmax = max(x1) ymin = min(x2) ymax = max(x2) if method == 'tricontourf': triang = tri.Triangulation(x1, x2) plt.tricontourf(triang, data, *args, **kwargs) elif method == 'scipy_griddata_pcontourf': n = np.sqrt(len(data)) xi = np.linspace(xmin, xmax, n) yi = np.linspace(ymin, ymax, n) zi = sgriddata((x1, x2), q, (xi[None,:], yi[:,None]), method='nearest') cs = plt.contourf(xi, yi, zi, *args, **kwargs) elif method == 'matplotlib_griddata_pcontourf': n = np.sqrt(len(data)) xi = np.linspace(xmin, xmax, n) yi = np.linspace(ymin, ymax, n) zi = mgriddata(x1, x2, q, xi, yi, interp='linear') cs = plt.contourf(xi, yi, zi, *args, **kwargs)
def plot_tchiphase2(filename): ''' Plot Phase diagram of T*chi - 2D version. ''' #params Gamma=0.5/pi NS=50 #plot ion() fig=figure(figsize=(6,4)) data=loadtxt(filename) Umesh=data[:,0] edmesh=data[:,1] gmoU=Gamma/Umesh edoU=edmesh/Umesh+0.5 tchimesh=data[:,2] #interpolating X,Y=meshgrid(linspace(0,0.55,200),linspace(0,0.55,200),indexing='ij') #Z=mgriddata(gmoU,edoU,tchimesh,X,Y,interp='linear') Z=mgriddata(gmoU,edoU,tchimesh,Y,X,interp='nn') pcolormesh(X,Y,Z,rasterized=True) pcolormesh(-X,Y,Z,rasterized=True) xlabel('$\\tilde{\epsilon}_f/U$',fontsize=18) xticks([-0.5,0,0.5],fontsize=14) ylabel('$\Gamma/U$',fontsize=18) yticks([0,0.5],fontsize=14) xlim(-0.5,0.5) ylim(0,0.55) cb=colorbar() cb.set_ticks([0,0.25]) text(0.,0.2,'LM',fontsize=18,color='w',horizontalalignment='center') text(0.35,0.4,'ASC',fontsize=18,color='w',horizontalalignment='center') tight_layout() pdb.set_trace() savefig('%s.eps'%filename[:-4]) savefig('%s.png'%filename[:-4])
def plot_Fig15(filename1,filename2,filename11,filename22): ''' Plot Phase diagram of T*chi - 2D version-two in one version. ''' nzcomp=[0,1,3] which='s' #params D=1. Gap=0.1 #plot gs=GridSpec(2,11,hspace=0.01) ymax=0.2 ymin=-0.15 if which=='d' else -0.2 ion() fig=figure(figsize=(9,6)) fig1=subplot(gs[0,:5]) colormap=cm.rainbow(linspace(0.2,0.8,len(nzcomp))) plts=[] data=loadtxt(filename1) wlist=data[:,0] dlv0=data[:,1:5]/pi dlv=data[:,5:]/pi for color,i in zip(colormap,nzcomp): mask1=wlist<-Gap mask2=wlist>Gap plts+=plot(concatenate([wlist[mask1],[-Gap,Gap],wlist[mask2]]),concatenate([dlv0[mask1,i],zeros(2),dlv0[mask2,i]]),lw=2,color=color) for color,i in zip(colormap,nzcomp): #scatter(wlist[mask],dlv[mask,i],s=30,lw=1,edgecolors=colormap[i],facecolors='none') plts.append(scatter(wlist[::3],dlv[::3,i],s=30,lw=1,edgecolors=color,facecolors='none')) legend(plts,[r'$d_0$',r'$d_x$',r'$d_z$',r"$d'_0$",r"$d'_x$",r"$d'_z$"],ncol=2,loc=1 if which=='d' else 3,fontsize=12) ylim(ymin,ymax) yticks([-0.2,-0.1,0,0.1,0.2]) xlim(-1,1) xlabel(r'$\omega/\omega_U$',fontsize=16) xticks([-1,0,1],fontsize=14) text(0.8,0.17,'(a)',fontsize=14) axhline(y=0,color='#999999',ls=':',lw=2) paramstring=r'\usepackage{bm}' matplotlib.rcParams['text.latex.preamble'] = paramstring matplotlib.rcParams['text.usetex'] = True #ylabel(r'$\bm{\mathcal{D}}(\omega)$',fontsize=16) ylabel(r'$\mathcal{D}(\omega)$',fontsize=16) xlabel(r'$\omega$/\omega_U',fontsize=16) xticks([-1,0,1],fontsize=14) ylim(ymin,ymax) xlim(-1,1) text(0.8,0.17,'(a)',fontsize=14) axhline(y=0,color='#999999',ls=':',lw=2) fig2=subplot(gs[0,5:10]) data=loadtxt(filename2) wlist=data[:,0] dlv0=data[:,1:5] dlv=data[:,5:] for color,i in zip(colormap,nzcomp): mask1=wlist<-Gap mask2=wlist>Gap plot(concatenate([wlist[mask1],[-Gap,Gap],wlist[mask2]]),concatenate([dlv0[mask1,i],zeros(2),dlv0[mask2,i]]),lw=2,color=color) for color,i in zip(colormap,nzcomp): mask1=wlist<-Gap mask2=wlist>Gap mask1[1::2]=0 mask2[0::2]=0 scatter(wlist[mask1|mask2],dlv[mask1|mask2,i],s=30,lw=1,edgecolors=color,facecolors='none') ylim(ymin,ymax) xlim(-1,1) xlabel(r'$\omega$',fontsize=16) xticks([-1,0,1],fontsize=14) yticks([]) text(0.8,0.17,'(b)',fontsize=14) axhline(y=0,color='#999999',ls=':',lw=2) #params Gamma=0.5/pi NS=50 #plot fig1=subplot(gs[1,:5]) text(0.48,0.2,'(c)',fontsize=14,color='w') data=loadtxt(filename11) Umesh=data[:,0] edmesh=data[:,1] gmoU=Gamma/Umesh edoU=edmesh/Umesh+0.5 tchimesh=data[:,2] #interpolating X,Y=meshgrid(linspace(-0.6,0.6,200),linspace(0,1/4.5,200),indexing='ij') Z=mgriddata(gmoU,edoU,tchimesh,Y,X,interp='nn') pcolormesh(X,Y,Z,rasterized=True) plot([0,0],[0,0.3],ls='--',color='k') xlabel('$\epsilon_f/U$',fontsize=16) xticks([-0.5,-0.3,0,0.3,0.5]) ylabel('$\Gamma/U$',fontsize=16) #yticks([0,0.5],fontsize=14) xlim(-0.6,0.6) ylim(0,1/4.5) text(0.,0.1,'doublet',fontsize=18,color='w',horizontalalignment='center') text(-0.4,0.2,'singlet',fontsize=18,color='w',horizontalalignment='center') fig2=subplot(gs[1,5:]) text(0.48,0.2,'(d)',fontsize=14,color='w') data=loadtxt(filename22) Umesh=data[:,0] edmesh=data[:,1] gmoU=Gamma/Umesh edoU=edmesh/Umesh+0.5 tchimesh=data[:,2] #interpolating Z=mgriddata(gmoU,edoU,tchimesh,Y,X,interp='nn') pcolormesh(X,Y,Z,rasterized=True) xlabel('$\epsilon_f/U$',fontsize=18) yticks([]) xlim(-0.6,0.6) xticks([-0.5,-0.3,0,0.3,0.5]) ylim(0,1/4.5) plot([0,0.43],[0,0.5],ls='--',color='k') #cax = fig.add_axes([0.9, 0.1, 0.03, 0.8]) #cb=fig.colorbar(img,cax=cax) cb=colorbar() cb.set_ticks([0,0.25]) text(0.,0.1,'doublet',fontsize=18,color='w',horizontalalignment='center') text(-0.4,0.2,'singlet',fontsize=18,color='w',horizontalalignment='center') y0=1./2/pi plot([-0.25],[y0],'wo') offsetx=0.03 plot([0.25],[y0],'wo') text(0.25+offsetx,y0,'D',fontsize=14,color='w') text(-0.25+offsetx,y0,'S',fontsize=14,color='w') gs.tight_layout(fig,pad=0.3,w_pad=0.6) pdb.set_trace() savefig('allinone.eps') savefig('allinone.png')
def plot_Fig5(filename1,filename2): ''' Plot Phase diagram of T*chi - 2D version-two in one version. ''' #params Gamma=0.5/pi NS=50 #plot ion() fig=figure(figsize=(8,3.5)) gs=GridSpec(1,11,hspace=0.01) fig1=subplot(gs[:5]) text(0.4,0.2,'(a)',fontsize=14,color='w') data=loadtxt(filename1) Umesh=data[:,0] edmesh=data[:,1] gmoU=Gamma/Umesh edoU=edmesh/Umesh+0.5 tchimesh=data[:,2] #interpolating X,Y=meshgrid(linspace(-0.6,0.6,200),linspace(0,1/4.5,200),indexing='ij') Z=mgriddata(gmoU,edoU,tchimesh,Y,X,interp='nn') pcolormesh(X,Y,Z,rasterized=True) plot([0,0],[0,0.3],ls='--',color='k') xlabel('$\epsilon_f/U$',fontsize=16) xticks([-0.5,-0.3,0,0.3,0.5]) ylabel('$\Gamma/U$',fontsize=16) #yticks([0,0.5],fontsize=14) xlim(-0.6,0.6) ylim(0,1/4.5) text(0.,0.1,'doublet',fontsize=18,color='w',horizontalalignment='center') text(-0.4,0.2,'singlet',fontsize=18,color='w',horizontalalignment='center') #y0=1./2/pi #xs,xd=0.35,0. #plot([xs],[y0],'wo') #offsetx=0.03 #plot([xd],[y0],'wo') #text(xd+offsetx,y0,'D1',fontsize=14,color='w') #text(xs+offsetx,y0,'S1',fontsize=14,color='w') fig2=subplot(gs[5:]) text(0.4,0.2,'(b)',fontsize=14,color='w') data=loadtxt(filename2) Umesh=data[:,0] edmesh=data[:,1] gmoU=Gamma/Umesh edoU=edmesh/Umesh+0.5 tchimesh=data[:,2] #interpolating Z=mgriddata(gmoU,edoU,tchimesh,Y,X,interp='nn') pcolormesh(X,Y,Z,rasterized=True) xlabel('$\epsilon_f/U$',fontsize=18) yticks([]) xlim(-0.6,0.6) xticks([-0.5,-0.3,0,0.3,0.5]) ylim(0,1/4.5) plot([0,0.43],[0,0.5],ls='--',color='k') #cax = fig.add_axes([0.9, 0.1, 0.03, 0.8]) #cb=fig.colorbar(img,cax=cax) cb=colorbar() cb.set_ticks([0,0.25]) text(0.,0.1,'doublet',fontsize=18,color='w',horizontalalignment='center') text(-0.4,0.2,'singlet',fontsize=18,color='w',horizontalalignment='center') y0=1./2/pi #plot([-0.25],[y0],'wo') plot([0],[y0],'wo') offsetx=0.03 text(0.+offsetx,y0,'D',fontsize=14,color='w') #text(-0.25+offsetx,y0,'S2',fontsize=14,color='w') gs.tight_layout(fig,pad=0.3,w_pad=0.6) pdb.set_trace() savefig('tchiskew.eps') savefig('tchiskew.png')
def plot_Fig4(filename1,filename2): ''' Plot Phase diagram of T*chi - 2D version-two in one version. ''' #params Gamma=0.5/pi NS=50 #plot ion() fig=figure(figsize=(8,4)) gs=GridSpec(1,11) subplot(gs[:5]) text(0.4,0.5,'(a)',fontsize=14,color='w') data=loadtxt(filename1) Umesh=data[:,0] edmesh=data[:,1] gmoU=Gamma/Umesh edoU=edmesh/Umesh+0.5 tchimesh=data[:,2] #interpolating X,Y=meshgrid(linspace(-0.55,0.55,200),linspace(0,0.55,200),indexing='ij') gmoU=append(gmoU,gmoU);edoU=append(-edoU,edoU);tchimesh=append(tchimesh,tchimesh) #Z=mgriddata(gmoU,edoU,tchimesh,X,Y,interp='linear') Z=mgriddata(gmoU,edoU,tchimesh,Y,X,interp='nn') pcolormesh(X,Y,Z,rasterized=True) #pcolormesh(-X,Y,Z,rasterized=True) xlabel('$\epsilon_f/U$',fontsize=16) #xticks([-0.5,0,0.5],fontsize=14) ylabel('$\Gamma/U$',fontsize=16) #yticks([0,0.5],fontsize=14) xlim(-0.5,0.5) ylim(0,0.55) text(0.,0.2,'doublet',fontsize=18,color='w',horizontalalignment='center') text(0.35,0.43,'singlet',fontsize=18,color='w',horizontalalignment='center') subplot(gs[5:]) text(0.4,0.5,'(b)',fontsize=14,color='w') data=loadtxt(filename2) Umesh=data[:,0] edmesh=data[:,1] gmoU=Gamma/Umesh edoU=edmesh/Umesh+0.5 tchimesh=data[:,2] #interpolating gmoU=append(gmoU,gmoU);edoU=append(-edoU,edoU);tchimesh=append(tchimesh,tchimesh) Z=mgriddata(gmoU,edoU,tchimesh,Y,X,interp='nn') pcolormesh(X,Y,Z,rasterized=True) #img=pcolormesh(-X,Y,Z,rasterized=True) xlabel('$\epsilon_f/U$',fontsize=18) #xticks([-0.5,0,0.5],fontsize=14) yticks([]) xlim(-0.5,0.5) ylim(0,0.55) #cax = fig.add_axes([0.9, 0.1, 0.03, 0.8]) #cb=fig.colorbar(img,cax=cax) cb=colorbar() cb.set_ticks([0,0.25]) text(0.,0.2,'doublet',fontsize=18,color='w',horizontalalignment='center') text(0.35,0.43,'singlet',fontsize=18,color='w',horizontalalignment='center') tight_layout() pdb.set_trace() savefig('tchi2.eps') savefig('tchi2.png')