simplify

Author: jreback

pandas/core/groupby/groupby.py

@@ -4740,10 +4740,7 @@ def _wrap_transformed_output(self, output, names=None):
 
     def _wrap_agged_blocks(self, items, blocks):
         if not self.as_index:
-            if blocks[0].values.ndim > 1:
-                index = np.arange(blocks[0].values.shape[1])
-            else:
-                index = np.arange(blocks[0].values.shape[0])
+            index = np.arange(blocks[0].values.shape[-1])
             mgr = BlockManager(blocks, [items, index])
             result = DataFrame(mgr)