ray.data.expressions.ColumnExpr.power

ColumnExpr.power(exponent: Any) → UDFExpr

Raise the expression to the given power.

Parameters:

exponent – The exponent to raise the expression to.

Returns:

A UDFExpr that computes the power operation.

Example

>>> from ray.data.expressions import col, lit
>>> import ray
>>> ds = ray.data.from_items([{"x": 2}, {"x": 3}])
>>> ds = ds.with_column("x_squared", col("x").power(2))
>>> # Result: x_squared = [4, 9]
>>> ds = ds.with_column("x_cubed", col("x").power(3))
>>> # Result: x_cubed = [8, 27]