From 11b23127eefad95050491a54a63a9e6ae4d6daa1 Mon Sep 17 00:00:00 2001 From: ahmedk <karim.ahmed@xfel.eu> Date: Thu, 21 Jul 2022 16:07:26 +0200 Subject: [PATCH] keep nan in plot for bad pixel code fig --- notebooks/LPD/LPDChar_Darks_NBC.ipynb | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/notebooks/LPD/LPDChar_Darks_NBC.ipynb b/notebooks/LPD/LPDChar_Darks_NBC.ipynb index 1122d39ea..277d5231c 100644 --- a/notebooks/LPD/LPDChar_Darks_NBC.ipynb +++ b/notebooks/LPD/LPDChar_Darks_NBC.ipynb @@ -722,6 +722,8 @@ " title = '{} value'.format(const)\n", " if const == 'BadPixelsDark':\n", " vmax = 4\n", + " bpix_code = data.astype(np.float32)\n", + " bpix_code[bpix_code == 0] = np.nan\n", " title = 'Bad pixel code'\n", " label = title\n", "\n", @@ -730,12 +732,12 @@ " '3 {}'.format(BadPixels.OFFSET_OUT_OF_THRESHOLD.name),\n", " '4 {}'.format('MIXED')]\n", "\n", - " heatmapPlot(data, add_panels=False, cmap='viridis',\n", + " heatmapPlot(bpix_code, add_panels=False, cmap='viridis',\n", " y_label='Rows', x_label='Columns',\n", " lut_label='', vmax=vmax,\n", " use_axis=ax, cb_ticklabels=cb_labels, cb_ticks = np.arange(4)+1,\n", " title='{}'.format(title))\n", - "\n", + " del bpix_code\n", " else:\n", "\n", " heatmapPlot(data, add_panels=False, cmap='viridis',\n", @@ -1157,7 +1159,7 @@ " l_data.append(np.count_nonzero(badpix_g[cap][qm][:,:,:,gain] & bit.value))\n", " \n", " if old_const[cap][qm]['BadPixelsDark'] is not None:\n", - " old_const[cap][qm]['BadPixelsDark'] = old_const[cap][qm]['BadPixelsDark'].astype(np.float32)\n", + " old_const[cap][qm]['BadPixelsDark'] = old_const[cap][qm]['BadPixelsDark'].astype(np.uint32)\n", " dataold = np.copy(old_const[cap][qm]['BadPixelsDark'][:, :, :, gain])\n", " l_data_old.append(len(dataold[dataold>0].flatten()))\n", " for bit in bits:\n", -- GitLab