BUG: Fix DateFrameGroupBy.mean error for Int64 dtype

doc/source/whatsnew/v1.0.2.rst

@@ -80,6 +80,7 @@ Bug fixes
 - Fixed bug in setting values using a slice indexer with string dtype (:issue:`31772`)
 - Fixed bug where :meth:`GroupBy.first` and :meth:`GroupBy.last` would raise a ``TypeError`` when groups contained ``pd.NA`` in a column of object dtype (:issue:`32123`)
 - Fix bug in :meth:`Series.convert_dtypes` for series with mix of integers and strings (:issue:`32117`)
+- Fixed bug where :meth:`DataFrameGroupBy.mean` would raise a ``TypeError`` on ``Int64`` dtype columns (:issue:`32219`)
 
 .. ---------------------------------------------------------------------------
 

pandas/core/groupby/generic.py

@@ -1083,7 +1083,7 @@ def _cython_agg_blocks(
                         result = type(block.values)._from_sequence(
                             result.ravel(), dtype=block.values.dtype
                         )
-                    except ValueError:
+                    except (ValueError, TypeError):
                         # reshape to be valid for non-Extension Block
                         result = result.reshape(1, -1)
 

pandas/tests/groupby/test_apply.py

@@ -866,3 +866,24 @@ def fct(group):
         [[1.0, 2.0], [3.0], [np.nan]], index=pd.Index(["a", "b", "none"], name="A")
     )
     tm.assert_series_equal(result, expected)
+
+
+@pytest.mark.parametrize("contains_na", [True, False])
+def test_apply_mean_with_nullable_integer(contains_na):
+    # https://github.com/pandas-dev/pandas/issues/32219
+    if contains_na:
+        values = {
+            "a": [1, 1, 1, 2, 2, 2, 3, 3, 3],
+            "b": [1, pd.NA, 2, 1, pd.NA, 2, 1, pd.NA, 2],
+        }
+    else:
+        values = {"a": [1, 1, 2, 2, 3, 3], "b": [1, 2, 1, 2, 1, 2]}
+
+    groups = pd.DataFrame(values, dtype="Int64").groupby("a")
+    result = groups.mean()
+
+    idx = pd.Index([1, 2, 3], dtype=object, name="a")
+    arr = np.array([1.5] * 3, dtype=float)
+    expected = pd.DataFrame({"b": arr}, index=idx)
+
+    tm.assert_frame_equal(result, expected)