def wrap_results_for_axis(self):
results = self.results<EOL>if self.result_type == '<STR_LIT>':<EOL><INDENT>result = self.infer_to_same_shape()<EOL><DEDENT>elif not isinstance(results[<NUM_LIT:0>], ABCSeries):<EOL><INDENT>from pandas import Series<EOL>result = Series(results)<EOL>result.index = self.res_index<EOL><DEDENT>else:<EOL><INDENT>result = self.infer_to_same_shape()<EOL><DEDENT>return result<EOL>
return the results for the columns