oemof / oemof-solph

        return [node_names.get(n, None) for n in names]

def node_weight_by_type(results, node_type):

    """

    Extracts node weights (if exist) of all components of the specified

    `node_type`.

    Node weight are endogenous optimzation variables associated with the node

    and not the edge between two node, foxample the variable representing the

    storage level.

    Parameters

    ----------

    results: dict

        A result dictionary from a solved oemof.solph.Model object

    node_type: oemof.solph class

        Specifies the type for which node weights should be collected,

        e.g. solph.components.GenericStorage

    Example

    --------

    ::

        from oemof.solph import views

        # solve oemof model 'm'

        # Then collect node weights

        views.node_weight_by_type(

            m.results(),

           node_type=solph.components.GenericStorage

        )

    """

    group = {

        k: v["sequences"]

        for k, v in results.items()

        if isinstance(k[0], node_type) and k[1] is None

    }

    if not group:

        logging.error(

            "No node weights for nodes of type `{}`".format(node_type)

        return df

def node_input_by_type(results, node_type, droplevel=None):

    """Gets all inputs for all nodes of the type `node_type` and returns

    a dataframe.

    Parameters

    ----------

    results: dict

        A result dictionary from a solved oemof.solph.Model object

    node_type: oemof.solph class

        Specifies the type of the node for that inputs are selected,

        e.g. solph.components.Sink

    droplevel: list

    Examples

    -----

    ::

        from oemof import solph

        from oemof.solph import views

        # solve oemof solph model 'm'

        # Then collect node weights

        views.node_input_by_type(

            m.results(),

            node_type=solph.components.Sink

        )

    """

    if droplevel is None:

        droplevel = []

    group = {

        k: v["sequences"]

        for k, v in results.items()

        if isinstance(k[1], node_type) and k[0] is not None

    }

        return df

def node_output_by_type(results, node_type, droplevel=None):

    """Gets all outputs for all nodes of the type `node_type` and returns

    a dataframe.

    Parameters

    ----------

    results: dict

        A result dictionary from a solved oemof.solph.Model object

    node_type: oemof.solph class

        Specifies the type of the node for that outputs are selected,

        e.g. solph.components.Converter

    droplevel: list

    Examples

    --------

    ::

        import oemof.solph as solph

        from oemof.solph import views

        # solve oemof solph model 'm'

        # Then collect node weights

        views.node_output_by_type(

            m.results(),

            node_type=solph.components.Converter

        )

    """

    if droplevel is None:

        droplevel = []

    group = {

        k: v["sequences"]

        for k, v in results.items()

        if isinstance(k[0], node_type) and k[1] is not None

    }