Merge pull request #554 from Quentin62/fpcaplot

fix IndexError in FPCAPlot
GAA-UAM · Jul 6, 2023 · 415796b · 415796b
2 parents 48fb0ad + 4a56dd7
commit 415796b
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/skfda/exploratory/visualization/fpca.py b/skfda/exploratory/visualization/fpca.py
@@ -84,9 +84,10 @@ def _plot(
             self.mean = self.mean.mean()
 
         for i, ax in enumerate(axes):
-            perturbations = self._get_component_perturbations(i)
-            GraphPlot(fdata=perturbations, axes=ax).plot()
-            ax.set_title(f"Principal component {i + 1}")
+            if i < self.n_subplots:
+                perturbations = self._get_component_perturbations(i)
+                GraphPlot(fdata=perturbations, axes=ax).plot()
+                ax.set_title(f"Principal component {i + 1}")
 
     def _get_component_perturbations(self, index: int = 0) -> FData:
         """