fix(plot): use nanmin and nanmax to get axes lims

cemsbv · Jul 23, 2024 · 3ed032b · 3ed032b
1 parent 5aabb8d
commit 3ed032b
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/baec/measurements/settlement_rod_measurement_series.py b/src/baec/measurements/settlement_rod_measurement_series.py
@@ -3,6 +3,7 @@
 from functools import cache
 from typing import List
 
+import numpy as np
 import pandas as pd
 from matplotlib import pyplot as plt
 from matplotlib.axes import Axes
@@ -263,7 +264,9 @@ def plot_z_time(self, axes: Axes | None = None) -> Axes:
             legend=True,
         )
 
-        axes.set_ylim(df[z_cols].values.min() - 0.5, df[z_cols].values.max() + 0.5)
+        axes.set_ylim(
+            np.nanmin(df[z_cols].values) - 0.5, np.nanmax(df[z_cols].values) + 0.5
+        )
         axes.grid()
 
         axes.set_ylabel(self.coordinate_reference_systems.vertical_datum_and_units)