Source code for xpark.dataset.namespace_expressions.map_namespace

"""Map namespace for expression operations on map-typed columns."""

from __future__ import annotations

from dataclasses import dataclass

from ray.data.namespace_expressions.map_namespace import _MapNamespace as RayMapNamespace




@dataclass
class _MapNamespace(RayMapNamespace):
    """Namespace for map operations on expression columns.

    This namespace provides methods for operating on map-typed columns
    (including MapArrays and ListArrays of Structs) using PyArrow UDFs.

    Example:
        >>> from xpark.dataset.expressions import col
        >>> # Get keys from map column
        >>> expr = col("headers").map.keys()
        >>> # Get values from map column
        >>> expr = col("headers").map.values()
    """