causaldag.classes.dag.DAG.children_of

DAG.children_of(nodes: Union[Hashable, Set[Hashable]]) → Set[Hashable][source]

Return all nodes that are children of the node or set of nodes nodes.

Parameters:nodes – A node or set of nodes.

Examples

>>> import causaldag as cd
>>> g = cd.DAG(arcs={(1, 2), (2, 3)})
>>> g.children_of(1)
{2}
>>> g.children_of({1, 2})
{2, 3}