def METHOD_NAME(*, values, radius, k1, k2, k3, r1, r2):
    for i in prange(len(values)):  # pylint: disable=not-an-iterable
        if radius[i] < r1:
            values[i] = k1 * radius[i] ** 2
        elif radius[i] < r2:
            values[i] = k2 * radius[i]
        else:
            values[i] = k3 * radius[i] ** (1 / 2)