Make reduce return single block

Author: mroeschke

pandas/core/internals/blocks.py

@@ -350,7 +350,7 @@ def apply(self, func, **kwargs) -> list[Block]:
         return self._split_op_result(result)
 
     @final
-    def reduce(self, func) -> list[Block]:
+    def reduce(self, func) -> Block:
         # We will apply the function and reshape the result into a single-row
         #  Block with the same mgr_locs; squeezing will be done at a higher level
         assert self.ndim == 2
@@ -362,8 +362,7 @@ def reduce(self, func) -> list[Block]:
         else:
             res_values = result.reshape(-1, 1)
 
-        nb = self.make_block(res_values)
-        return [nb]
+        return self.make_block(res_values)
 
     @final
     def _split_op_result(self, result: ArrayLike) -> list[Block]:

pandas/core/internals/managers.py

@@ -1651,11 +1651,7 @@ def reduce(self, func: Callable) -> Self:
         # If 2D, we assume that we're operating column-wise
         assert self.ndim == 2
 
-        res_blocks: list[Block] = []
-        for blk in self.blocks:
-            nbs = blk.reduce(func)
-            res_blocks.extend(nbs)
-
+        res_blocks = [blk.reduce(func) for blk in self.blocks]
         index = default_index(1)  # placeholder
         new_mgr = type(self).from_blocks(res_blocks, [self.items, index])
         return new_mgr