Source code for open_cge.household

def eqF(beta, py, Y, pf):
    '''
    Factor demand.

    .. math::
        F_{h,j} = \\beta_{h,j}\\frac{py_{j}}{pf_{h}}Y_{j}

    Args:
        beta (2D numpy array): Cost share parameter for factor h in
            production of good j
        py (1D array): The price of value added for each good j
        Y (1D numpy array): Value added for each good j
        pf (1D array): Price of each factor h

    Returns:
        F (2D numpy array): The demand for factor h used in the
            production of good j
    '''
    F = beta.div(pf, axis=0) * Y * py
    return F


def eqI(pf, Ff, Sp, Td, Fsh, Trf):
    '''
    Total income of consumers.

    .. math::
        I = \left(\sum_{h}pf_{h}Ff_{h} - S^{p} - T^{d}- FSH - TRF\\right)

    Args:
        pf (1D numpy array): The price of factor h
        Ff (1D numpy array): Endowment of factor h
        Sp (float): Total household savings
        Td (float): Total direct tax revenue
        Fsh = Repatriated profits
        Trf (float): Total transfers to households

    Returns:
        I (float): Total income of consumers
    '''
    I = (pf * Ff).sum() - Sp - Td - Fsh + Trf
    return I


def eqXp(alpha, I, pq):
    '''
    Demand for production good i by consumers.

    .. math::
        X^{p}_{i}= \\frac{\\alpha_{i}}{pq_{i}}I

    Args:
        alpha (1D numpy array): Budget share of good i
        I (float): Total income of consumers
        pq (1D numpy array): price of the Armington good (domestic +
            imports) for each good i

    Returns:
        Xp (1D numpy array): Demand for production good i by consumers
    '''
    Xp = alpha.div(pq, axis=0) * I
    return Xp