From 0cc4a1f215adc68b19cc5918e08dddf89a065f2e Mon Sep 17 00:00:00 2001 From: Florez Ospina Juan Felipe Date: Mon, 3 Jun 2024 14:17:42 +0200 Subject: [PATCH] Updated treemap visualization to select only root metadata, which is of string type. --- src/hdf5_vis.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/src/hdf5_vis.py b/src/hdf5_vis.py index e771b30..40c8977 100644 --- a/src/hdf5_vis.py +++ b/src/hdf5_vis.py @@ -28,8 +28,15 @@ def display_group_hierarchy_on_a_treemap(filename: str): metadata_dict={} for key in file.attrs.keys(): #if 'metadata' in key: - metadata_dict[key[key.find('_')+1::]]= file.attrs[key] - metadata_list.append(key[key.find('_')+1::]+':'+file.attrs[key]) + if isinstance(file.attrs[key], str): # Check if the attribute is a string + metadata_key = key[key.find('_') + 1:] + metadata_value = file.attrs[key] + metadata_dict[metadata_key] = metadata_value + metadata_list.append(f'{metadata_key}: {metadata_value}') + + #metadata_dict[key[key.find('_')+1::]]= file.attrs[key] + #metadata_list.append(key[key.find('_')+1::]+':'+file.attrs[key]) + metadata = '
'.join(['
'] + metadata_list) customdata_series = pd.Series(nodes)