def plot_2Dpmf(X, Y, xlabel, ylabel, title, saveas, bins=(100, 100)):
cubecmap.set_bad(color="lightgray", alpha=1.)
plt.figure()
z, x, y = np.histogram2d(X, Y, bins=bins)
F = -np.log(z)
F -= F.min()
extent = [x[0], x[-1], y[0], y[-1]]
F = np.ma.masked_invalid(F)
ax1 = plt.contourf(F.T, 100, extent=extent, cmap=cubecmap)
plt.xlabel(xlabel, fontsize=16)
plt.ylabel(ylabel, fontsize=16)
cbar = plt.colorbar()
cbar.ax.set_ylabel('Free energy (k$_B$T)', fontsize=15)
plt.title(title, fontsize=16)
if saveas is not None:
plt.savefig(saveas + ".png", bbox_inches="tight")
plt.savefig(saveas + ".pdf", bbox_inches="tight")
plt.savefig(saveas + ".eps", bbox_inches="tight")
def plot_2Dpmf(X,Y,xlabel,ylabel,title,saveas,bins=(100,100)):
cubecmap.set_bad(color="lightgray",alpha=1.)
plt.figure()
z,x,y = np.histogram2d(X,Y,bins=bins)
F = -np.log(z)
F -= F.min()
extent = [x[0], x[-1], y[0], y[-1]]
F = np.ma.masked_invalid(F)
ax1 = plt.contourf(F.T,100,extent=extent,cmap=cubecmap)
plt.xlabel(xlabel,fontsize=16)
plt.ylabel(ylabel,fontsize=16)
cbar = plt.colorbar()
cbar.ax.set_ylabel('Free energy (k$_B$T)',fontsize=15)
plt.title(title,fontsize=16)
if saveas is not None:
plt.savefig(saveas+".png",bbox_inches="tight")
plt.savefig(saveas+".pdf",bbox_inches="tight")
plt.savefig(saveas+".eps",bbox_inches="tight")
def plot_weights(C, coordname, title, saveas, shift):
C = np.ma.masked_invalid(C)
cubecmap.set_bad(color="lightgray", alpha=1.)
rwb = plt.get_cmap("bwr_r")
rwb.set_bad(color="lightgray", alpha=1.)
#shifted_cmap = shiftedColorMap(rwb, midpoint=shift, name='shifted')
plt.figure()
#plt.pcolormesh(C,cmap=cubecmap)
plt.pcolormesh(C, cmap=rwb, vmin=-0.1, vmax=0.1)
cbar = plt.colorbar()
cbar.ax.set_ylabel('Contact weight on %s' % coordname, fontsize=15)
plt.xlabel("Residue i", fontsize=16)
plt.ylabel("Residue j", fontsize=16)
plt.xlim(0, C.shape[0])
plt.ylim(0, C.shape[0])
plt.title(title, fontsize=16)
if saveas is not None:
plt.savefig(saveas + ".png", bbox_inches="tight")
plt.savefig(saveas + ".pdf", bbox_inches="tight")
plt.savefig(saveas + ".eps", bbox_inches="tight")
def plot_weights(C,coordname,title,saveas,shift):
C = np.ma.masked_invalid(C)
cubecmap.set_bad(color="lightgray",alpha=1.)
rwb = plt.get_cmap("bwr_r")
rwb.set_bad(color="lightgray",alpha=1.)
#shifted_cmap = shiftedColorMap(rwb, midpoint=shift, name='shifted')
plt.figure()
#plt.pcolormesh(C,cmap=cubecmap)
plt.pcolormesh(C,cmap=rwb,vmin=-0.1,vmax=0.1)
cbar = plt.colorbar()
cbar.ax.set_ylabel('Contact weight on %s' % coordname,fontsize=15)
plt.xlabel("Residue i",fontsize=16)
plt.ylabel("Residue j",fontsize=16)
plt.xlim(0,C.shape[0])
plt.ylim(0,C.shape[0])
plt.title(title,fontsize=16)
if saveas is not None:
plt.savefig(saveas+".png",bbox_inches="tight")
plt.savefig(saveas+".pdf",bbox_inches="tight")
plt.savefig(saveas+".eps",bbox_inches="tight")
