BUG: array-like quantile fails on column groupby

doc/source/whatsnew/v1.2.0.rst

@@ -734,6 +734,7 @@ Groupby/resample/rolling
 - Bug in :meth:`DataFrame.groupby` dropped ``nan`` groups from result with ``dropna=False`` when grouping over a single column (:issue:`35646`, :issue:`35542`)
 - Bug in :meth:`.DataFrameGroupBy.head`, :meth:`.DataFrameGroupBy.tail`, :meth:`SeriesGroupBy.head`, and :meth:`SeriesGroupBy.tail` would raise when used with ``axis=1`` (:issue:`9772`)
 - Bug in :meth:`.DataFrameGroupBy.transform` would raise when used with ``axis=1`` and a transformation kernel (e.g. "shift") (:issue:`36308`)
+- Bug in :meth:`.DataFrameGroupBy.quantile` couldn't handle with arraylike ``q`` when grouping by columns (:issue:`35269`)
 
 Reshaping
 ^^^^^^^^^

pandas/core/groupby/groupby.py

@@ -2227,6 +2227,7 @@ def post_processor(vals: np.ndarray, inference: Optional[Type]) -> np.ndarray:
  )
  for qi in q
  ]
+ if self.axis == 0:
  result = concat(results, axis=0, keys=q)
  # fix levels to place quantiles on the inside
  # TODO(GH-10710): Ideally, we could write this as
@@ -2250,6 +2251,22 @@ def post_processor(vals: np.ndarray, inference: Optional[Type]) -> np.ndarray:
  # reorder rows to keep things sorted
  indices = np.arange(len(result)).reshape([len(q), self.ngroups]).T.flatten()
  return result.take(indices)
+ else:
+ result = concat(results, axis=1, keys=q)
+
+ order = list(range(1, result.columns.nlevels)) + [0]
+ index_names = np.array(result.columns.names)
+ result.columns.names = np.arange(len(index_names))
+ result = result.reorder_levels(order, axis=1)
+ result.columns.names = index_names[order]
+ indices = (
+ np.arange(result.shape[1])
+ .reshape(
+ [len(q), self.ngroups],
+ )
+ .T.flatten()
+ )
+ return result.take(indices, axis=1)
 
  @Substitution(name="groupby")
  def ngroup(self, ascending: bool = True):

pandas/tests/groupby/test_quantile.py

@@ -254,3 +254,26 @@ def test_groupby_timedelta_quantile():
  index=Index([1, 2], name="group"),
  )
  tm.assert_frame_equal(result, expected)
+
+
+def test_columns_groupby_quantile():
+ # GH 33795
+ df = DataFrame(
+ np.arange(12).reshape(3, -1),
+ index=list("XYZ"),
+ columns=pd.Series(list("ABAB"), name="col"),
+ )
+ result = df.groupby("col", axis=1).quantile(q=[0.8, 0.2])
+ expected = DataFrame(
+ [
+ [1.6, 0.4, 2.6, 1.4],
+ [5.6, 4.4, 6.6, 5.4],
+ [9.6, 8.4, 10.6, 9.4],
+ ],
+ index=list("XYZ"),
+ columns=Index(
+ [("A", 0.8), ("A", 0.2), ("B", 0.8), ("B", 0.2)], names=["col", None]
+ ),
+ )
+
+ tm.assert_frame_equal(result, expected)