Gravitational accelleration

gravitational_accelleration.py

def g_at_distance(g_base, radius, distance):
    """
    Calculate gravitational accelleration at a distance
    - above the surface G * (radius / distance)^2
    - below the surface G * (distance / radius)
    
    Parameters
    ----------
    g_base : float
        gravitational accelleration at the surface
    radius : int, float
        surface radius
    distance : ndarray
        array of distances from the center of mass
        
    Returns
    -------
    g : ndarray
        Gravitational accelleration at a distance
    """
    above = np.divide(radius, distance, where=distance != 0) ** 2
    below = (distance / radius)
    g = g_base * np.where(distance < radius, below, above)
    return g