Function where

xarray/core/duck_array_ops.py:407–419

Three argument where() with better dtype promotion rules.

(condition, x, y)

def where(condition, x, y):
    """Three argument where() with better dtype promotion rules."""
    xp = get_array_namespace(condition, x, y)

    dtype = xp.bool_ if hasattr(xp, "bool_") else xp.bool
    if not is_duck_array(condition):
        condition = asarray(condition, dtype=dtype, xp=xp)
    else:
        condition = astype(condition, dtype=dtype, xp=xp)

    promoted_x, promoted_y = as_shared_dtype([x, y], xp=xp)

    return xp.where(condition, promoted_x, promoted_y)


def where_method(data, cond, other=dtypes.NA):