daft.functions.image_channel#

def image_channel(image: Expression) -> Expression:
    """Gets the number of channels in an image.

    Args:
        image (Image Expression): image to retrieve the number of channels from.

    Returns:
        Expression (UInt32 Expression): An Expression representing the number of channels in the image.

    Example:
        >>> import daft
        >>> from daft.functions import image_channel
        >>> # Create a dataframe with an image column
        >>> df = ...  # doctest: +SKIP
        >>> df = df.with_column("channel", image_channel(df["images"]))  # doctest: +SKIP
    """
    return image_attribute(image, "channel")