Skip to content

Commit 4dad9a9

Browse files
VincentStimperVincentStimper
committed
Colormap updated
1 parent 42a0f91 commit 4dad9a9

4 files changed

Lines changed: 29 additions & 17 deletions

File tree

‎figures/Fig2_Theory_vs_Reconstruction.ipynb‎

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,9 @@
4141
"mpl.rcParams['pdf.fonttype'] = 42\n",
4242
"mpl.rcParams['ps.fonttype'] = 42\n",
4343
"\n",
44+
"colornames = ['#646464', '#666666', '#6a6a6a', '#6f6f6f', '#737373', '#787878', '#7d7d7d', '#828282', '#878787', '#8d8d8d', '#929292', '#989898', '#9e9e9e', '#a4a4a4', '#aaaaaa', '#b0b0b0', '#b6b6b6', '#bcbcbc', '#c2c2c2', '#c9c9c9', '#cfcfcf', '#d6d6d6', '#dcdcdc', '#e3e3e3', '#eaeaea', '#efefee', '#efeee5', '#efeddc', '#efecd3', '#eeebca', '#eeeac0', '#eee9b7', '#eee8ad', '#ede7a4', '#ede69a', '#ede590', '#ede487', '#ece37d', '#ece273', '#ece069', '#ecdf5f', '#ebde55', '#ebdd4b', '#ebdc41', '#ebdb37', '#ebd333', '#ebc933', '#ecbe32', '#ecb432', '#eda931', '#ee9e31', '#ee9330', '#ef8830', '#ef7d2f', '#f0722f', '#f0672e', '#f15c2e', '#f2512d', '#f2462d', '#f33b2c', '#f3302c', '#f4252b', '#f4192b', '#ef182f', '#e81834', '#e21939', '#db1a3e', '#d51a43', '#ce1b48', '#c71b4d', '#c11c52', '#ba1c58', '#b31d5d', '#ac1d62', '#a61e67', '#9f1e6c', '#981f72', '#911f77', '#8a207c', '#842182']\n",
45+
"custom_cmap = mpl.colors.LinearSegmentedColormap.from_list('custom', colornames, N=256)\n",
46+
"\n",
4447
"# Create plot folder if needed\n",
4548
"if not os.path.exists('../results/figures'):\n",
4649
" os.mkdir('../results/figures')"
@@ -111,7 +114,7 @@
111114
" kyy = np.reshape(kyy, (nkx * nky,))\n",
112115
" Ek_scaled = np.reshape(intFunc(np.column_stack((kxx, kyy))), (nkx, nky))\n",
113116
" \n",
114-
" axs[i,0].imshow(Ek_scaled.T*msk, cmap='Spectral_r',\n",
117+
" axs[i,0].imshow(Ek_scaled.T*msk, cmap=custom_cmap,\n",
115118
" extent=[kx[0]+kxshift, kx[-1]+kxshift, ky[0], ky[-1]])\n",
116119
" axs[i,0].spines['right'].set_color('none')\n",
117120
" axs[i,0].spines['top'].set_color('none')\n",
@@ -130,7 +133,7 @@
130133
" axs[i,0].plot(kpks[:,1]-0.02, kpks[:,0], '--k', lw=1, dashes=[5, 4])\n",
131134
" \n",
132135
" # Draw the reconstruction side\n",
133-
" axs[i,1].imshow(recbands[ind,...]*msk, cmap='Spectral_r', extent=[kx[0]+kxshift, kx[-1]+kxshift, ky[0], ky[-1]])\n",
136+
" axs[i,1].imshow(recbands[ind,...]*msk, cmap=custom_cmap, extent=[kx[0]+kxshift, kx[-1]+kxshift, ky[0], ky[-1]])\n",
134137
" axs[i,1].spines['right'].set_color('none')\n",
135138
" axs[i,1].spines['top'].set_color('none')\n",
136139
" axs[i,1].tick_params(axis='both', length=8, width=2, labelsize=28)\n",
@@ -174,7 +177,7 @@
174177
" kyy = np.reshape(kyy, (nkx * nky,))\n",
175178
" Ek_scaled = np.reshape(intFunc(np.column_stack((kxx, kyy))), (nkx, nky))\n",
176179
" \n",
177-
" axs[i,0].imshow(Ek_scaled.T*msk, cmap='Spectral_r',\n",
180+
" axs[i,0].imshow(Ek_scaled.T*msk, cmap=custom_cmap,\n",
178181
" extent=[kx[0]+0.05, kx[-1]+0.05, ky[0], ky[-1]])\n",
179182
" axs[i,0].spines['right'].set_color('none')\n",
180183
" axs[i,0].spines['top'].set_color('none')\n",
@@ -193,7 +196,7 @@
193196
" axs[i,0].plot(kpks[:,1]-0.02, kpks[:,0], '--k', lw=1, dashes=[5, 4])\n",
194197
" \n",
195198
" # Draw the reconstruction side\n",
196-
" axs[i,1].imshow(recbands[ind,...]*msk, cmap='Spectral_r', extent=[kx[0]+kxshift, kx[-1]+kxshift, ky[0], ky[-1]])\n",
199+
" axs[i,1].imshow(recbands[ind,...]*msk, cmap=custom_cmap, extent=[kx[0]+kxshift, kx[-1]+kxshift, ky[0], ky[-1]])\n",
197200
" axs[i,1].spines['right'].set_color('none')\n",
198201
" axs[i,1].spines['top'].set_color('none')\n",
199202
" axs[i,1].tick_params(axis='both', length=8, width=2, labelsize=28)\n",

‎figures/Fig3_HexagonalZernike.ipynb‎

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,6 +42,9 @@
4242
"mpl.rcParams['pdf.fonttype'] = 42\n",
4343
"mpl.rcParams['ps.fonttype'] = 42\n",
4444
"\n",
45+
"colornames = ['#646464', '#666666', '#6a6a6a', '#6f6f6f', '#737373', '#787878', '#7d7d7d', '#828282', '#878787', '#8d8d8d', '#929292', '#989898', '#9e9e9e', '#a4a4a4', '#aaaaaa', '#b0b0b0', '#b6b6b6', '#bcbcbc', '#c2c2c2', '#c9c9c9', '#cfcfcf', '#d6d6d6', '#dcdcdc', '#e3e3e3', '#eaeaea', '#efefee', '#efeee5', '#efeddc', '#efecd3', '#eeebca', '#eeeac0', '#eee9b7', '#eee8ad', '#ede7a4', '#ede69a', '#ede590', '#ede487', '#ece37d', '#ece273', '#ece069', '#ecdf5f', '#ebde55', '#ebdd4b', '#ebdc41', '#ebdb37', '#ebd333', '#ebc933', '#ecbe32', '#ecb432', '#eda931', '#ee9e31', '#ee9330', '#ef8830', '#ef7d2f', '#f0722f', '#f0672e', '#f15c2e', '#f2512d', '#f2462d', '#f33b2c', '#f3302c', '#f4252b', '#f4192b', '#ef182f', '#e81834', '#e21939', '#db1a3e', '#d51a43', '#ce1b48', '#c71b4d', '#c11c52', '#ba1c58', '#b31d5d', '#ac1d62', '#a61e67', '#9f1e6c', '#981f72', '#911f77', '#8a207c', '#842182']\n",
46+
"custom_cmap = mpl.colors.LinearSegmentedColormap.from_list('custom', colornames, N=256)\n",
47+
"\n",
4548
"# Create plot folder if needed\n",
4649
"if not os.path.exists('../results/figures'):\n",
4750
" os.mkdir('../results/figures')"
@@ -77,7 +80,7 @@
7780
"ff, axs = plt.subplots(5, 1, figsize=(3, 10))\n",
7881
"\n",
7982
"for ind, ibs in enumerate([3, 10, 27, 41, 89]):\n",
80-
" im = axs[ind].imshow(basis[ibs,...]*bmask[...], cmap='Spectral_r', vmin=-1.8, vmax=1.8)\n",
83+
" im = axs[ind].imshow(basis[ibs,...]*bmask[...], cmap=custom_cmap, vmin=-1.8, vmax=1.8)\n",
8184
" axs[ind].axis('off')\n",
8285
" axs[ind].text(5, 5, str(ind+1), fontsize=15, fontname=\"Arial\")\n",
8386
"\n",

‎figures/Fig5_K_and_Mprime.ipynb‎

Lines changed: 13 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,9 @@
3737
"mpl.rcParams['pdf.fonttype'] = 42\n",
3838
"mpl.rcParams['ps.fonttype'] = 42\n",
3939
"\n",
40+
"colornames = ['#646464', '#666666', '#6a6a6a', '#6f6f6f', '#737373', '#787878', '#7d7d7d', '#828282', '#878787', '#8d8d8d', '#929292', '#989898', '#9e9e9e', '#a4a4a4', '#aaaaaa', '#b0b0b0', '#b6b6b6', '#bcbcbc', '#c2c2c2', '#c9c9c9', '#cfcfcf', '#d6d6d6', '#dcdcdc', '#e3e3e3', '#eaeaea', '#efefee', '#efeee5', '#efeddc', '#efecd3', '#eeebca', '#eeeac0', '#eee9b7', '#eee8ad', '#ede7a4', '#ede69a', '#ede590', '#ede487', '#ece37d', '#ece273', '#ece069', '#ecdf5f', '#ebde55', '#ebdd4b', '#ebdc41', '#ebdb37', '#ebd333', '#ebc933', '#ecbe32', '#ecb432', '#eda931', '#ee9e31', '#ee9330', '#ef8830', '#ef7d2f', '#f0722f', '#f0672e', '#f15c2e', '#f2512d', '#f2462d', '#f33b2c', '#f3302c', '#f4252b', '#f4192b', '#ef182f', '#e81834', '#e21939', '#db1a3e', '#d51a43', '#ce1b48', '#c71b4d', '#c11c52', '#ba1c58', '#b31d5d', '#ac1d62', '#a61e67', '#9f1e6c', '#981f72', '#911f77', '#8a207c', '#842182']\n",
41+
"custom_cmap = mpl.colors.LinearSegmentedColormap.from_list('custom', colornames, N=256)\n",
42+
"\n",
4043
"# Create plot folder if needed\n",
4144
"if not os.path.exists('../results/figures'):\n",
4245
" os.mkdir('../results/figures')"
@@ -102,18 +105,18 @@
102105
" ax[1, i].set_yticks([])\n",
103106
" ax[1, i].set_xlabel('$k_x$ ($\\mathrm{\\AA}^{-1}$)', fontsize=15)\n",
104107
" \n",
105-
"cs00 = ax[0, 0].contourf(kxx, kyy, mpatch[0,...], 20, cmap='Spectral_r', vmin=-1.6, vmax=-1)\n",
108+
"cs00 = ax[0, 0].contourf(kxx, kyy, mpatch[0,...], 20, cmap=custom_cmap, vmin=-1.6, vmax=-1)\n",
106109
"ax[0, 0].contour(cs00, colors='k', linestyles='-', linewidths=0.5)\n",
107110
"ax[0, 0].text(0.65, 0.9, 'Band #1', fontsize=15, transform=ax[0,0].transAxes)\n",
108111
"\n",
109-
"cs10 = ax[1, 0].contourf(kxx, kyy, mpatch[1,:,:], 20, cmap='Spectral_r', vmin=-1.95, vmax=-1.4)\n",
112+
"cs10 = ax[1, 0].contourf(kxx, kyy, mpatch[1,:,:], 20, cmap=custom_cmap, vmin=-1.95, vmax=-1.4)\n",
110113
"ax[1, 0].contour(cs10, colors='k', linestyles='-', linewidths=0.5)\n",
111114
"ax[1, 0].text(0.65, 0.9, 'Band #2', fontsize=15, transform=ax[1,0].transAxes)\n",
112115
" \n",
113-
"cs01 = ax[0, 1].contourf(kxx, kyy, bands[0,...], 20, cmap='Spectral_r', vmin=-1.6, vmax=-1)\n",
116+
"cs01 = ax[0, 1].contourf(kxx, kyy, bands[0,...], 20, cmap=custom_cmap, vmin=-1.6, vmax=-1)\n",
114117
"ax[0, 1].contour(cs01, colors='k', linestyles='-', linewidths=0.5)\n",
115118
"\n",
116-
"cs11 = ax[1, 1].contourf(kxx, kyy, bands[1,...], 20, cmap='Spectral_r', vmin=-1.95, vmax=-1.4)\n",
119+
"cs11 = ax[1, 1].contourf(kxx, kyy, bands[1,...], 20, cmap=custom_cmap, vmin=-1.95, vmax=-1.4)\n",
117120
"ax[1, 1].contour(cs11, colors='k', linestyles='-', linewidths=0.5)\n",
118121
"\n",
119122
"plt.subplots_adjust(hspace=0.08, wspace=0.08)\n",
@@ -180,24 +183,24 @@
180183
" ax[1, i].set_yticks([])\n",
181184
" ax[1, i].set_xlabel('$k_x$ ($\\mathrm{\\AA}^{-1}$)', fontsize=15)\n",
182185
" \n",
183-
"cs00 = ax[0, 0].contourf(kxx, kyy, rcfb['bands'][0,ofs+1:-ofs+1,ofs:-ofs], 23, cmap='Spectral_r', vmax=-0.7, vmin=-1.2+minsft)\n",
186+
"cs00 = ax[0, 0].contourf(kxx, kyy, rcfb['bands'][0,ofs+1:-ofs+1,ofs:-ofs], 23, cmap=custom_cmap, vmax=-0.7, vmin=-1.2+minsft)\n",
184187
"ax[0, 0].contour(cs00, colors='k', linestyles='-', linewidths=0.5)\n",
185188
"ax[0, 0].text(0.65, 0.9, 'Band #1', fontsize=15, transform=ax[0,0].transAxes)\n",
186189
"\n",
187-
"cs10 = ax[1, 0].contourf(kxx, kyy, rcfb['bands'][1,ofs+1:-ofs+1,ofs:-ofs], 23, cmap='Spectral_r', vmax=-1.15, vmin=-1.55+minsft)\n",
190+
"cs10 = ax[1, 0].contourf(kxx, kyy, rcfb['bands'][1,ofs+1:-ofs+1,ofs:-ofs], 23, cmap=custom_cmap, vmax=-1.15, vmin=-1.55+minsft)\n",
188191
"ax[1, 0].contour(cs10, colors='k', linestyles='-', linewidths=0.5)\n",
189192
"ax[1, 0].text(0.65, 0.9, 'Band #2', fontsize=15, transform=ax[1,0].transAxes)\n",
190193
" \n",
191-
"cs01 = ax[0, 1].contourf(kxx, kyy, rfb['bands'][0,ofs+1:-ofs+1,ofs:-ofs], 23, cmap='Spectral_r', vmax=-0.7, vmin=-1.2+minsft)\n",
194+
"cs01 = ax[0, 1].contourf(kxx, kyy, rfb['bands'][0,ofs+1:-ofs+1,ofs:-ofs], 23, cmap=custom_cmap, vmax=-0.7, vmin=-1.2+minsft)\n",
192195
"ax[0, 1].contour(cs01, colors='k', linestyles='-', linewidths=0.5)\n",
193196
"\n",
194-
"cs11 = ax[1, 1].contourf(kxx, kyy, rfb['bands'][1,ofs+1:-ofs+1,ofs:-ofs], 23, cmap='Spectral_r', vmax=-1.15, vmin=-1.55+minsft)\n",
197+
"cs11 = ax[1, 1].contourf(kxx, kyy, rfb['bands'][1,ofs+1:-ofs+1,ofs:-ofs], 23, cmap=custom_cmap, vmax=-1.15, vmin=-1.55+minsft)\n",
195198
"ax[1, 1].contour(cs11, colors='k', linestyles='-', linewidths=0.5)\n",
196199
"\n",
197-
"cs02 = ax[0, 2].contourf(kxx, kyy, fitbs['b1'], 23, cmap='Spectral_r', vmax=-0.7, vmin=-1.2+minsft)\n",
200+
"cs02 = ax[0, 2].contourf(kxx, kyy, fitbs['b1'], 23, cmap=custom_cmap, vmax=-0.7, vmin=-1.2+minsft)\n",
198201
"ax[0, 2].contour(cs02, colors='k', linestyles='-', linewidths=0.5)\n",
199202
"\n",
200-
"cs12 = ax[1, 2].contourf(kxx, kyy, fitbs['b2'], 23, cmap='Spectral_r', vmax=-1.15, vmin=-1.55+minsft)\n",
203+
"cs12 = ax[1, 2].contourf(kxx, kyy, fitbs['b2'], 23, cmap=custom_cmap, vmax=-1.15, vmin=-1.55+minsft)\n",
201204
"ax[1, 2].contour(cs12, colors='k', linestyles='-', linewidths=0.5)\n",
202205
"\n",
203206
"cs03 = ax[0, 3].contourf(kxx, kyy, 1e3*(fitbs['b1'] - rfb['bands'][0,ofs+1:-ofs+1,ofs:-ofs]), 23, cmap='RdBu_r', vmax=12, vmin=-12)\n",

‎figures/SFig4_Hyperparameter_tuning.ipynb‎

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -42,6 +42,9 @@
4242
"mpl.rcParams['pdf.fonttype'] = 42\n",
4343
"mpl.rcParams['ps.fonttype'] = 42\n",
4444
"\n",
45+
"colornames = ['#646464', '#666666', '#6a6a6a', '#6f6f6f', '#737373', '#787878', '#7d7d7d', '#828282', '#878787', '#8d8d8d', '#929292', '#989898', '#9e9e9e', '#a4a4a4', '#aaaaaa', '#b0b0b0', '#b6b6b6', '#bcbcbc', '#c2c2c2', '#c9c9c9', '#cfcfcf', '#d6d6d6', '#dcdcdc', '#e3e3e3', '#eaeaea', '#efefee', '#efeee5', '#efeddc', '#efecd3', '#eeebca', '#eeeac0', '#eee9b7', '#eee8ad', '#ede7a4', '#ede69a', '#ede590', '#ede487', '#ece37d', '#ece273', '#ece069', '#ecdf5f', '#ebde55', '#ebdd4b', '#ebdc41', '#ebdb37', '#ebd333', '#ebc933', '#ecbe32', '#ecb432', '#eda931', '#ee9e31', '#ee9330', '#ef8830', '#ef7d2f', '#f0722f', '#f0672e', '#f15c2e', '#f2512d', '#f2462d', '#f33b2c', '#f3302c', '#f4252b', '#f4192b', '#ef182f', '#e81834', '#e21939', '#db1a3e', '#d51a43', '#ce1b48', '#c71b4d', '#c11c52', '#ba1c58', '#b31d5d', '#ac1d62', '#a61e67', '#9f1e6c', '#981f72', '#911f77', '#8a207c', '#842182']\n",
46+
"custom_cmap = mpl.colors.LinearSegmentedColormap.from_list('custom', colornames, N=256)\n",
47+
"\n",
4548
"# Create plot folder if needed\n",
4649
"if not os.path.exists('../results/figures'):\n",
4750
" os.mkdir('../results/figures')"
@@ -109,7 +112,7 @@
109112
"source": [
110113
"# Plot a grid of reconstructed bands\n",
111114
"ims, axs = vis.sliceview3d(smaband*msk[None,...], axis=0, ncol=7, vmin=-1, vmax=0,\n",
112-
" colormap='Spectral_r', axisreturn='nested', numbered=False, figsize=(20, 20*5/7),\n",
115+
" colormap=custom_cmap, axisreturn='nested', numbered=False, figsize=(20, 20*5/7),\n",
113116
" imkwds={'extent':[kx[0]+0.05, kx[-1]+0.05, ky[0]-0.02, ky[-1]-0.02]});\n",
114117
"\n",
115118
"# Annotate the plot-frames of good reconstructions\n",
@@ -165,7 +168,7 @@
165168
"fig = plt.figure(figsize=(8, 3))\n",
166169
"ax = fig.add_axes([0.05, 0.80, 0.04, 0.5])\n",
167170
"norm = mpl.colors.Normalize(vmin=-1, vmax=0)\n",
168-
"cb = mpl.colorbar.ColorbarBase(ax, cmap=plt.get_cmap('Spectral_r'),\n",
171+
"cb = mpl.colorbar.ColorbarBase(ax, cmap=custom_cmap,\n",
169172
" norm=norm, ticks=[-1, 0],\n",
170173
" orientation='vertical')\n",
171174
"# cb.ax.set_yticks([-1, 0])\n",

0 commit comments

Comments
 (0)