Return a numpy array when selecting a single column.

dukebody · dukebody · commit 31714cc1e38b · 2015-08-09T18:09:22.000+02:00
diff --git a/sklearn_pandas/__init__.py b/sklearn_pandas/__init__.py
@@ -95,7 +95,7 @@ def _get_col_subset(self, X, cols):
             X = X.df
 
         if return_vector:
-            t = X[cols[0]]
+            t = X[cols[0]].values
         else:
             t = X.as_matrix(cols)
 
