Fix plotting of just the relevant part of the spectrum

StingraySoftware · Sep 26, 2023 · f9f1421 · f9f1421
1 parent d2df1ee
commit f9f1421
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/stingray/crossspectrum.py b/stingray/crossspectrum.py
@@ -1076,8 +1076,11 @@ def plot(
             ax = fig.add_subplot(1, 1, 1)
 
         ax.plot(self.freq, np.abs(self.power), marker, color="b", label="Amplitude")
-        ax.plot(self.freq, self.power.real, marker, color="r", alpha=0.5, label="Real Part")
-        ax.plot(self.freq, self.power.imag, marker, color="g", alpha=0.5, label="Imaginary Part")
+        if np.any(np.iscomplex(self.power.imag)):
+            ax.plot(
+                self.freq, self.power.imag, marker, color="g", alpha=0.5, label="Imaginary Part"
+            )
+            ax.plot(self.freq, self.power.real, marker, color="r", alpha=0.5, label="Real Part")
 
         if labels is not None:
             try: