chore(rust): move for_each to polars-ops

crates/polars-core/src/chunked_array/ops/mod.rs

@@ -26,7 +26,6 @@ mod explode_and_offsets;
 mod extend;
 mod fill_null;
 mod filter;
-mod for_each;
 pub mod full;
 pub mod gather;
 #[cfg(feature = "interpolate")]

crates/polars-ops/src/chunked_array/for_each.rs

@@ -0,0 +1,11 @@
+use polars_core::prelude::*;
+
+pub fn for_each<'a, T, F>(ca: &'a ChunkedArray<T>, mut op: F)
+where
+    T: PolarsDataType,
+    F: FnMut(Option<T::Physical<'a>>),
+{
+    ca.downcast_iter().for_each(|arr| {
+        arr.iter().for_each(&mut op);
+    })
+}

crates/polars-ops/src/chunked_array/mod.rs

@@ -23,9 +23,12 @@ pub mod gather_skip_nulls;
 #[cfg(feature = "repeat_by")]
 mod repeat_by;
 
+pub mod for_each;
+
 pub use binary::*;
 #[cfg(feature = "timezones")]
 pub use datetime::*;
+pub use for_each::*;
 #[cfg(feature = "interpolate")]
 pub use interpolate::*;
 pub use list::*;

crates/polars-ops/src/chunked_array/strings/split.rs

@@ -3,6 +3,7 @@ use polars_arrow::export::arrow::array::{MutableArray, MutableUtf8Array};
 use polars_core::chunked_array::ops::arity::binary_elementwise_for_each;
 
 use super::*;
+use crate::prelude::for_each;
 
 #[cfg(feature = "dtype-struct")]
 pub fn split_to_struct<'a, F, I>(
@@ -21,7 +22,7 @@ where
 
     if by.len() == 1 {
         if let Some(by) = by.get(0) {
-            ca.for_each(|opt_s| match opt_s {
+            for_each(ca, |opt_s| match opt_s {
                 None => {
                     for arr in &mut arrs {
                         arr.push_null()
@@ -86,7 +87,7 @@ where
             let mut builder =
                 ListUtf8ChunkedBuilder::new(ca.name(), ca.len(), ca.get_values_size());
 
-            ca.for_each(|opt_s| match opt_s {
+            for_each(ca, |opt_s| match opt_s {
                 Some(s) => {
                     let iter = op(s, by);
                     builder.append_values_iter(iter)