Add list_transform

vortex-array/src/expr/exprs.rs

@@ -45,6 +45,8 @@ use crate::scalar_fn::fns::list_contains::ListContains;
 use crate::scalar_fn::fns::list_contains::ListContainsOptions;
 use crate::scalar_fn::fns::list_length::ListLength;
 use crate::scalar_fn::fns::list_sum::ListSum;
+use crate::scalar_fn::fns::list_transform::ListLambdaOptions;
+use crate::scalar_fn::fns::list_transform::ListTransform;
 use crate::scalar_fn::fns::literal::Literal;
 use crate::scalar_fn::fns::mask::Mask;
 use crate::scalar_fn::fns::merge::DuplicateHandling;
@@ -1262,6 +1264,17 @@ pub fn list_sum(input: Expression) -> Expression {
     ListSum.new_expr(NumericalAggregateOpts::default(), [input])
 }
 
+/// Creates an expression that transforms every element of a list with `lambda`.
+///
+/// `lambda` takes the element and, optionally, its `u64` index.
+pub fn list_transform(list: Expression, lambda: Lambda) -> VortexResult<Expression> {
+    Expression::try_new_higher_order(
+        ListTransform.bind(ListLambdaOptions::default()),
+        [list],
+        [lambda],
+    )
+}
+
 /// Creates a bound expression that sums the elements of each list.
 pub fn bound_list_sum(input: BoundExpression) -> BoundExpression {
     ListSum

vortex-array/src/expr/mod.rs

@@ -93,6 +93,7 @@ pub use exprs::list_contains_opts;
 pub use exprs::list_length;
 pub use exprs::list_sum;
 pub use exprs::list_sum_opts;
+pub use exprs::list_transform;
 pub use exprs::lit;
 pub use exprs::lt;
 pub use exprs::lt_eq;