Move method specialization inside subclass

Author: howsiwei

pandas/core/indexing.py

@@ -166,37 +166,7 @@ def _get_loc(self, key: int, axis: int):
     def _slice(self, obj, axis: int, kind=None):
         return self.obj._slice(obj, axis=axis, kind=kind)
 
-    def _ensure_listlike_indexer(self, key, axis: int):
-        """
-        Ensure that a list-like of keys are all present by adding them if they
-        do not already exist.
-
-        Parameters
-        ----------
-        key : list-like
-            Target labels
-        axis : int
-            Dimension on which the indexing is being made
-        """
-        if not isinstance(self.obj._get_axis(axis), ABCMultiIndex) and all(
-            is_hashable(k) for k in key
-        ):
-            for k in key:
-                try:
-                    self.obj[k]
-                except KeyError:
-                    self.obj[k] = np.nan
-
     def _get_setitem_indexer(self, key):
-        if (
-            self.name == "loc"  # column is indexed by name
-            and isinstance(key, tuple)
-            and len(key) >= 2  # key is at least 2-dimensional
-            and is_list_like_indexer(key[1])  # key indexes multiple columns
-            and not com.is_bool_indexer(key[1])
-        ):
-            self._ensure_listlike_indexer(key[1], axis=1)
-
         if self.axis is not None:
             return self._convert_tuple(key)
 
@@ -1841,6 +1811,36 @@ def _getitem_axis(self, key, axis: int):
         self._validate_key(key, axis)
         return self._get_label(key, axis=axis)
 
+    def _ensure_listlike_indexer(self, key, axis: int):
+        """
+        Ensure that a list-like of keys are all present by adding them if they
+        do not already exist.
+
+        Parameters
+        ----------
+        key : list-like
+            Target labels
+        axis : int
+            Dimension on which the indexing is being made
+        """
+        if not isinstance(self.obj._get_axis(axis), ABCMultiIndex) and all(
+            is_hashable(k) for k in key
+        ):
+            for k in key:
+                try:
+                    self.obj[k]
+                except KeyError:
+                    self.obj[k] = np.nan
+
+    def _get_setitem_indexer(self, key):
+        if (
+            isinstance(key, tuple)
+            and len(key) >= 2  # key is at least 2-dimensional
+            and is_list_like_indexer(key[1])  # key indexes multiple columns
+            and not com.is_bool_indexer(key[1])
+        ):
+            self._ensure_listlike_indexer(key[1], axis=1)
+        return super()._get_setitem_indexer(key)
 
 class _iLocIndexer(_LocationIndexer):
     """