Refactored if-else.

Author: rhshadrach

pandas/core/groupby/generic.py

@@ -1223,6 +1223,8 @@ def _wrap_applied_output(self, keys, values, not_indexed_same=False):
         if isinstance(first_not_none, DataFrame):
             return self._concat_objects(keys, values, not_indexed_same=not_indexed_same)
 
+        key_index = self.grouper.result_index if self.as_index else None
+
         if isinstance(first_not_none, NDFrame):
 
             # this is to silence a DeprecationWarning
@@ -1236,11 +1238,7 @@ def _wrap_applied_output(self, keys, values, not_indexed_same=False):
                 backup = first_not_none._constructor(**kwargs)
 
             values = [x if (x is not None) else backup for x in values]
-
-        key_index = self.grouper.result_index if self.as_index else None
-        v = values[0]
-
-        if not isinstance(v, (np.ndarray, Index, Series)):
+        else:
             # values are not series or array-like but scalars
             # self._selection_name not passed through to Series as the
             # result should not take the name of original selection
@@ -1252,6 +1250,8 @@ def _wrap_applied_output(self, keys, values, not_indexed_same=False):
                 self._insert_inaxis_grouper_inplace(result)
                 return result
 
+        v = values[0]
+
         if not isinstance(v, ABCSeries):
             # GH1738: values is list of arrays of unequal lengths
             # TODO: sure this is right?  we used to do this