Fix tranposition.

tjof2 · Sep 24, 2020 · 2150463 · 2150463
1 parent 4a223eb
commit 2150463
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/pguresvt/hspy.py b/pguresvt/hspy.py
@@ -68,7 +68,12 @@ def denoise(self, signal):
 
         super(HSPYSVT, self).denoise(self._X)
 
-        denoised_signal = signal._deepcopy_with_new_data(self.Y_.T)
+        if self._signal_type == "spectrum":
+            axes = (1, 0)
+        elif self._signal_type == "image":
+            axes = (2, 0, 1)
+
+        denoised_signal = signal._deepcopy_with_new_data(self.Y_.transpose(axes))
         new_title = f"Denoised {signal.metadata.General.title}".strip()
         denoised_signal.metadata.General.title = new_title