diff --git a/src/webactogram/webactogram.py b/src/webactogram/webactogram.py index 6a4c63a..43415c0 100644 --- a/src/webactogram/webactogram.py +++ b/src/webactogram/webactogram.py @@ -437,7 +437,8 @@ def apply_median_blurring(self): def define_pcolormesh_args(self): """ define the x, y and z (color) data structure for plotting later on """ - xx, yy, zz = self.act.dd, self.act.h2, np.tile(self.pcm['z'], (2, 1)) # tile the z array to create a 2D array + xx, yy, zz = self.act.dd, self.act.h2, self.pcm['z'] + zz = np.vstack((zz, np.pad(zz, ((0,0),(0,1)))[:, 1:])) if not self.act.landscape: # if vertical xx, yy = yy, xx # swap the x and y arrays