diff --git a/plots/triangular_sing/sc.py b/plots/triangular_sing/sc.py index f8ad755e8b980da400754d2fe5b91a803dde52d2..1c139f62059845506ec39b72cd1b7a2a3412ec5f 100755 --- a/plots/triangular_sing/sc.py +++ b/plots/triangular_sing/sc.py @@ -102,16 +102,20 @@ cmap = mpc.LinearSegmentedColormap('CustomColormap', custom_colors) phase = np.load('phase.npy') sing = np.load("sing.npy") +alphas = alphas[:15] +sing = sing[:, :15] +phase = phase[:, :15] + phase = np.array(phase, dtype='float') fig, axs = plt.subplot_mosaic([['SC', 'space1', 'cbar', 'space', 'S1', 'T11', 'space2', 'S2', 'T21'], ['SC', 'space1', 'cbar', 'space', 'T12', 'T13', 'space2', 'T22', 'T23']], - gridspec_kw={'width_ratios': [2.5, 0.2, - 0.32, 0.45, + gridspec_kw={'width_ratios': [4, 0.4, + 0.5, 0.55, 2, 2, 0.2, 2, 2]}, - figsize=(fig_width, 0.42 * fig_width)) + figsize=(fig_width, 0.40 * fig_width)) axs['space'].axis('off') axs['space2'].axis('off') axs['space1'].axis('off') @@ -133,7 +137,7 @@ axs['SC'].imshow(phase[:, ::-1].T, vmin=0, vmax=2, cmap='gray', **image_kw, axs['SC'].text(0.28, 0.5, '\slshape{}(i)SDW', ha='center', va='center', rotation=90) -cb = plt.colorbar(im, cax=axs['cbar']) +cb = plt.colorbar(im, cax=axs['cbar'], fraction=0.046, pad=0.4) cb.set_ticks([0, 1]) cb.set_label('singlet weight', labelpad=-8) axs['cbar'].yaxis.set_ticks_position('left') @@ -222,8 +226,6 @@ ev_colors = { vmax = np.max([np.max(pts[0].flatten().real), np.min(pts[0].flatten().imag), np.max(pts[1].flatten().real), np.max(pts[1].flatten().imag)]) -print(vmax) - ev_cmap = mpc.LinearSegmentedColormap('CustomColormap', ev_colors) for ii, nn in enumerate(set1): @@ -233,7 +235,7 @@ for ii, nn in enumerate(set1): ax.axis('off') ax.tricontourf(mesh_bf[:, 0], mesh_bf[:, 1], np.imag(pts[0][:, ii]), 20, vmin=-vmax, vmax=vmax, cmap=cmap) - ax.text(-4.2, 2.8, text[ii]) + ax.text(-4.5, 2.8, text[ii]) for ii, nn in enumerate(set2): ax = axs[nn] @@ -242,7 +244,7 @@ for ii, nn in enumerate(set2): ax.axis('off') ax.tricontourf(mesh_bf[:, 0], mesh_bf[:, 1], np.real(pts[1][:, ii]), 20, vmin=-vmax, vmax=vmax, cmap=cmap) - ax.text(-4.2, 2.8, text[ii]) + ax.text(-4.5, 2.8, text[ii]) fig.text(0.435, 0.06, r'$\Im(\text{State 1})$') fig.text(0.72, 0.06, r'$\Re(\text{State 2})$')