def where(boolean_expr, true_expr, false_null_expr):
"""
Equivalent to the ternary expression: if X then Y else Z
Parameters
----------
boolean_expr : BooleanValue (array or scalar)
true_expr : value
Values for each True value
false_null_expr : value
Values for False or NULL values
Returns
-------
result : arity depending on inputs
Type of true_expr used to determine output type
"""
op = _ops.Where(boolean_expr, true_expr, false_null_expr)
return op.to_expr()
def where(
boolean_expr: ir.BooleanValue,
true_expr: ir.ValueExpr,
false_null_expr: ir.ValueExpr,
) -> ir.ValueExpr:
"""Return `true_expr` if `boolean_expr` is `True` else `false_null_expr`.
Parameters
----------
boolean_expr
A boolean expression
true_expr
Value returned if `boolean_expr` is `True`
false_null_expr
Value returned if `boolean_expr` is `False` or `NULL`
Returns
-------
ir.ValueExpr
An expression
"""
op = ops.Where(boolean_expr, true_expr, false_null_expr)
return op.to_expr()