TYP: plotting

Author: jbrockmendel

pandas/plotting/_matplotlib/boxplot.py

@@ -93,17 +93,18 @@ def __init__(self, data, return_type: str = "axes", **kwargs) -> None:
     # error: Signature of "_plot" incompatible with supertype "MPLPlot"
     @classmethod
     def _plot(  # type: ignore[override]
-        cls, ax: Axes, y, column_num=None, return_type: str = "axes", **kwds
+        cls, ax: Axes, y: np.ndarray, column_num=None, return_type: str = "axes", **kwds
     ):
+        ys: np.ndarray | list[np.ndarray]
         if y.ndim == 2:
-            y = [remove_na_arraylike(v) for v in y]
+            ys = [remove_na_arraylike(v) for v in y]
             # Boxplot fails with empty arrays, so need to add a NaN
             #   if any cols are empty
             # GH 8181
-            y = [v if v.size > 0 else np.array([np.nan]) for v in y]
+            ys = [v if v.size > 0 else np.array([np.nan]) for v in ys]
         else:
-            y = remove_na_arraylike(y)
-        bp = ax.boxplot(y, **kwds)
+            ys = remove_na_arraylike(y)
+        bp = ax.boxplot(ys, **kwds)
 
         if return_type == "dict":
             return bp, bp

pandas/plotting/_matplotlib/core.py

@@ -1448,7 +1448,14 @@ def _make_plot(self, fig: Figure) -> None:
     # error: Signature of "_plot" incompatible with supertype "MPLPlot"
     @classmethod
     def _plot(  # type: ignore[override]
-        cls, ax: Axes, x, y, style=None, column_num=None, stacking_id=None, **kwds
+        cls,
+        ax: Axes,
+        x,
+        y: np.ndarray,
+        style=None,
+        column_num=None,
+        stacking_id=None,
+        **kwds,
     ):
         # column_num is used to get the target column from plotf in line and
         # area plots
@@ -1475,7 +1482,7 @@ def _ts_plot(self, ax: Axes, x, data: Series, style=None, **kwds):
             decorate_axes(ax.right_ax, freq, kwds)
         ax._plot_data.append((data, self._kind, kwds))
 
-        lines = self._plot(ax, data.index, data.values, style=style, **kwds)
+        lines = self._plot(ax, data.index, np.asarray(data.values), style=style, **kwds)
         # set date formatter, locators and rescale limits
         # error: Argument 3 to "format_dateaxis" has incompatible type "Index";
         # expected "DatetimeIndex | PeriodIndex"
@@ -1503,7 +1510,9 @@ def _initialize_stacker(cls, ax: Axes, stacking_id, n: int) -> None:
 
     @final
     @classmethod
-    def _get_stacked_values(cls, ax: Axes, stacking_id, values, label):
+    def _get_stacked_values(
+        cls, ax: Axes, stacking_id: int | None, values: np.ndarray, label
+    ) -> np.ndarray:
         if stacking_id is None:
             return values
         if not hasattr(ax, "_stacker_pos_prior"):
@@ -1523,7 +1532,7 @@ def _get_stacked_values(cls, ax: Axes, stacking_id, values, label):
 
     @final
     @classmethod
-    def _update_stacker(cls, ax: Axes, stacking_id, values) -> None:
+    def _update_stacker(cls, ax: Axes, stacking_id: int | None, values) -> None:
         if stacking_id is None:
             return
         if (values >= 0).all():
@@ -1601,7 +1610,7 @@ def _plot(  # type: ignore[override]
         cls,
         ax: Axes,
         x,
-        y,
+        y: np.ndarray,
         style=None,
         column_num=None,
         stacking_id=None,
@@ -1727,7 +1736,7 @@ def _plot(  # type: ignore[override]
         cls,
         ax: Axes,
         x,
-        y,
+        y: np.ndarray,
         w,
         start: int | npt.NDArray[np.intp] = 0,
         log: bool = False,
@@ -1860,7 +1869,7 @@ def _plot(  # type: ignore[override]
         cls,
         ax: Axes,
         x,
-        y,
+        y: np.ndarray,
         w,
         start: int | npt.NDArray[np.intp] = 0,
         log: bool = False,

pandas/plotting/_matplotlib/hist.py

@@ -101,7 +101,7 @@ def _calculate_bins(self, data: DataFrame, bins) -> np.ndarray:
     def _plot(  # type: ignore[override]
         cls,
         ax: Axes,
-        y,
+        y: np.ndarray,
         style=None,
         bottom: int | np.ndarray = 0,
         column_num: int = 0,
@@ -166,7 +166,7 @@ def _make_plot(self, fig: Figure) -> None:
 
             self._append_legend_handles_labels(artists[0], label)
 
-    def _make_plot_keywords(self, kwds: dict[str, Any], y) -> None:
+    def _make_plot_keywords(self, kwds: dict[str, Any], y: np.ndarray) -> None:
         """merge BoxPlot/KdePlot properties to passed kwds"""
         # y is required for KdePlot
         kwds["bottom"] = self.bottom
@@ -225,7 +225,7 @@ def __init__(
         self.weights = weights
 
     @staticmethod
-    def _get_ind(y, ind):
+    def _get_ind(y: np.ndarray, ind):
         if ind is None:
             # np.nanmax() and np.nanmin() ignores the missing values
             sample_range = np.nanmax(y) - np.nanmin(y)
@@ -248,12 +248,12 @@ def _get_ind(y, ind):
     def _plot(  #  type: ignore[override]
         cls,
         ax: Axes,
-        y,
+        y: np.ndarray,
         style=None,
         bw_method=None,
         ind=None,
         column_num=None,
-        stacking_id=None,
+        stacking_id: int | None = None,
         **kwds,
     ):
         from scipy.stats import gaussian_kde
@@ -265,7 +265,7 @@ def _plot(  #  type: ignore[override]
         lines = MPLPlot._plot(ax, ind, y, style=style, **kwds)
         return lines
 
-    def _make_plot_keywords(self, kwds: dict[str, Any], y) -> None:
+    def _make_plot_keywords(self, kwds: dict[str, Any], y: np.ndarray) -> None:
         kwds["bw_method"] = self.bw_method
         kwds["ind"] = self._get_ind(y, ind=self.ind)