BUG: Using boolean keys to select a column (GH44322)

Author: DriesSchaumont

doc/source/whatsnew/v1.4.0.rst

@@ -576,7 +576,7 @@ Indexing
 - Bug in :meth:`Series.__setitem__` with an integer dtype other than ``int64`` setting with a ``range`` object unnecessarily upcasting to ``int64`` (:issue:`44261`)
 - Bug in :meth:`Series.__setitem__` with a boolean mask indexer setting a listlike value of length 1 incorrectly broadcasting that value (:issue:`44265`)
 - Bug in :meth:`DataFrame.loc.__setitem__` and :meth:`DataFrame.iloc.__setitem__` with mixed dtypes sometimes failing to operate in-place (:issue:`44345`)
--
+- Bug in :meth:`DataFrame.loc.__getitem__` incorrectly raising ``KeyError`` when selecting a single column with a boolean key (:issue:`44322`).
 
 Missing
 ^^^^^^^

pandas/core/indexing.py

@@ -994,7 +994,7 @@ def _validate_key(self, key, axis: int):
         # slice of labels (where start-end in labels)
         # slice of integers (only if in the labels)
         # boolean not in slice and with boolean index
-        if isinstance(key, bool) and not is_bool_dtype(self.obj.index):
+        if isinstance(key, bool) and not is_bool_dtype(self.obj._get_axis(axis)):
             raise KeyError(
                 f"{key}: boolean label can not be used without a boolean index"
             )

pandas/tests/indexing/test_loc.py

@@ -178,6 +178,13 @@ def test_column_types_consistent(self):
         )
         tm.assert_frame_equal(df, expected)
 
+    def test_loc_getitem_column_boolean_arg(self):
+        # GH 44322
+        df = DataFrame([[1]], columns=Index([False]))
+        res = df.loc[:, False]
+        exp = Series([1], name=False)
+        tm.assert_series_equal(res, exp)
+
 
 class TestLoc2:
     # TODO: better name, just separating out things that rely on base class