feat(api): add zip operation and api

ibis/backends/base/sql/alchemy/datatypes.py

@@ -4,6 +4,7 @@
 
 import sqlalchemy as sa
 import sqlalchemy.types as sat
+import toolz
 from multipledispatch import Dispatcher
 from sqlalchemy.dialects import postgresql
 from sqlalchemy.engine.default import DefaultDialect

ibis/expr/operations/arrays.py

@@ -186,3 +186,21 @@ class ArrayUnion(Value):
 
     output_dtype = rlz.dtype_like("args")
     output_shape = rlz.shape_like("args")
+
+
+@public
+class Zip(Value):
+    arg = rlz.tuple_of(rlz.array, min_length=2)
+
+    output_shape = rlz.shape_like("arg")
+
+    @attribute.default
+    def output_dtype(self):
+        return dt.Array(
+            dt.Struct(
+                {
+                    f"f{i:d}": array.output_dtype.value_type
+                    for i, array in enumerate(self.arg, start=1)
+                }
+            )
+        )

ibis/expr/types/arrays.py

@@ -759,6 +759,24 @@ def union(self, other: ir.ArrayValue) -> ir.ArrayValue:
         """
         return ops.ArrayUnion(self, other).to_expr()
 
+    def zip(self, other: ir.Array, *others: ir.Array) -> ir.Array:
+        """Zip two or more arrays together.
+
+        Parameters
+        ----------
+        other
+            Another array to zip with `self`
+        others
+            Additional arrays to zip with `self`
+
+        Returns
+        -------
+        Array
+            Array of structs where each struct field is an element of each input
+            array.
+        """
+        return ops.Zip((self, other, *others)).to_expr()
+
 
 @public
 class ArrayScalar(Scalar, ArrayValue):