From 0d4e955eb0ef072b27870c6417f01e587bd3978a Mon Sep 17 00:00:00 2001 From: Grzegorz Kowal Date: Wed, 6 Oct 2021 18:02:16 -0300 Subject: [PATCH] PYTHON: Reduce dimensions for 2D datasets in AmunH5. Signed-off-by: Grzegorz Kowal --- python/amunpy/src/amunpy/amunh5.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/python/amunpy/src/amunpy/amunh5.py b/python/amunpy/src/amunpy/amunh5.py index 2bef2ee..f23a799 100644 --- a/python/amunpy/src/amunpy/amunh5.py +++ b/python/amunpy/src/amunpy/amunh5.py @@ -137,4 +137,8 @@ class AmunH5(Amun): cname = os.path.join(self.dirname, self.chunks[chunk_number]['filename']) with h5py.File(cname, 'r') as h5: - return numpy.array(h5['variables'][dataset_name]) + dset = numpy.array(h5['variables'][dataset_name]) + if self.attributes['ndims'] == 3: + return dset + else: + return dset[0,:,:,:]