Merge pull request #655 from ivg/fixes-get-cols

fixes get_cols to return properly chopped series
owlbarn · Mar 7, 2024 · 00fef9b · 00fef9b
2 parents abee52c + a26149b
commit 00fef9b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/base/misc/owl_dataframe.ml b/src/base/misc/owl_dataframe.ml
@@ -414,7 +414,7 @@ let set_by_name x i name a =
   set x i j a
 
 
-let to_cols x = x.data
+let to_cols x = Array.init (Array.length x.data) (get_col x)
 
 let to_rows x =
   let stack = Owl_utils.Stack.make () in