use ._should_fallback_to_positional

Author: Terji Petersen

pandas/plotting/_core.py

@@ -17,7 +17,6 @@
 
 from pandas.core.dtypes.common import (
     is_integer,
-    is_integer_dtype,
     is_list_like,
 )
 from pandas.core.dtypes.generic import (
@@ -941,15 +940,15 @@ def __call__(self, *args, **kwargs):
                         f"{kind} requires either y column or 'subplots=True'"
                     )
                 if y is not None:
-                    if is_integer(y) and not is_integer_dtype(data.columns):
+                    if is_integer(y) and data.columns._should_fallback_to_positional:
                         y = data.columns[y]
                     # converted to series actually. copy to not modify
                     data = data[y].copy()
                     data.index.name = y
         elif isinstance(data, ABCDataFrame):
             data_cols = data.columns
             if x is not None:
-                if is_integer(x) and not is_integer_dtype(data.columns):
+                if is_integer(x) and data.columns._should_fallback_to_positional:
                     x = data_cols[x]
                 elif not isinstance(data[x], ABCSeries):
                     raise ValueError("x must be a label or position")
@@ -958,7 +957,7 @@ def __call__(self, *args, **kwargs):
                 # check if we have y as int or list of ints
                 int_ylist = is_list_like(y) and all(is_integer(c) for c in y)
                 int_y_arg = is_integer(y) or int_ylist
-                if int_y_arg and not is_integer_dtype(data.columns):
+                if int_y_arg and data.columns._should_fallback_to_positional:
                     y = data_cols[y]
 
                 label_kw = kwargs["label"] if "label" in kwargs else False

pandas/plotting/_matplotlib/core.py

@@ -1137,9 +1137,9 @@ def __init__(self, data, x, y, **kwargs) -> None:
         MPLPlot.__init__(self, data, **kwargs)
         if x is None or y is None:
             raise ValueError(self._kind + " requires an x and y column")
-        if is_integer(x) and not is_integer_dtype(self.data.columns):
+        if is_integer(x) and self.data.columns._should_fallback_to_positional:
             x = self.data.columns[x]
-        if is_integer(y) and not is_integer_dtype(self.data.columns):
+        if is_integer(y) and self.data.columns._should_fallback_to_positional:
             y = self.data.columns[y]
 
         # Scatter plot allows to plot objects data
@@ -1196,7 +1196,7 @@ def __init__(self, data, x, y, s=None, c=None, **kwargs) -> None:
         elif is_hashable(s) and s in data.columns:
             s = data[s]
         super().__init__(data, x, y, s=s, **kwargs)
-        if is_integer(c) and not is_integer_dtype(self.data.columns):
+        if is_integer(c) and self.data.columns._should_fallback_to_positional:
             c = self.data.columns[c]
         self.c = c
 
@@ -1291,7 +1291,7 @@ def _kind(self) -> Literal["hexbin"]:
 
     def __init__(self, data, x, y, C=None, **kwargs) -> None:
         super().__init__(data, x, y, **kwargs)
-        if is_integer(C) and not is_integer_dtype(self.data.columns):
+        if is_integer(C) and self.data.columns._should_fallback_to_positional:
             C = self.data.columns[C]
         self.C = C