From 0c80dd5e41b4366d91132c8d19390ac3fb3a538a Mon Sep 17 00:00:00 2001
From: ahmedk <karim.ahmed@xfel.eu>
Date: Tue, 7 Nov 2023 13:36:43 +0100
Subject: [PATCH] fix strixel plotting
---
.../Jungfrau_Gain_Correct_and_Verify_NBC.ipynb | 15 ++++++++++-----
1 file changed, 10 insertions(+), 5 deletions(-)
diff --git a/notebooks/Jungfrau/Jungfrau_Gain_Correct_and_Verify_NBC.ipynb b/notebooks/Jungfrau/Jungfrau_Gain_Correct_and_Verify_NBC.ipynb
index a247e97ed..79f35f8a7 100644
--- a/notebooks/Jungfrau/Jungfrau_Gain_Correct_and_Verify_NBC.ipynb
+++ b/notebooks/Jungfrau/Jungfrau_Gain_Correct_and_Verify_NBC.ipynb
@@ -831,8 +831,9 @@
" **mean_plot_kwargs\n",
" )\n",
"else:\n",
- " ax.imshow(corrected_mean.squeeze(), aspect=aspect, **mean_plot_kwargs)\n",
- " \n",
+ " corr = ax.imshow(corrected_mean.squeeze(), aspect=aspect, **mean_plot_kwargs)\n",
+ " plt.colorbar(corr)\n",
+ "\n",
"ax.set_title(f'{karabo_id} - Mean CORRECTED', size=18)\n",
"\n",
"plt.show()"
@@ -858,7 +859,8 @@
" **mean_plot_kwargs\n",
" )\n",
"else:\n",
- " ax.imshow(corrected_mean.squeeze(), aspect=aspect, **mean_plot_kwargs)\n",
+ " corr = ax.imshow(corrected_masked_mean.squeeze(), aspect=aspect, **mean_plot_kwargs)\n",
+ " plt.colorbar(corr)\n",
"\n",
"ax.set_title(f'{karabo_id} - Mean CORRECTED with mask', size=18)\n",
"\n",
@@ -889,7 +891,8 @@
" **single_plot_kwargs\n",
" )\n",
"else:\n",
- " ax.imshow(corrected_train.squeeze(), aspect=aspect, **single_plot_kwargs)\n",
+ " corr = ax.imshow(corrected_train.squeeze(), aspect=aspect, **single_plot_kwargs)\n",
+ " plt.colorbar(corr)\n",
"\n",
"ax.set_title(f\"{karabo_id} - CORRECTED train: {tid}\", size=18)\n",
"\n",
@@ -1059,7 +1062,9 @@
" colorbar={'shrink': 1, 'pad': 0.01},\n",
" )\n",
"else:\n",
- " ax.imshow(np.log2(mask_train).squeeze(), vmin=0, vmax=32, cmap='jet', aspect=aspect)\n",
+ " mask = ax.imshow(\n",
+ " mask_train.squeeze(), vmin=0, vmax=32, cmap='jet', aspect=aspect)\n",
+ " plt.colorbar(mask)\n",
"\n",
"plt.show()"
]
--
GitLab