Skip to content

Add nodes from dict #762

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
nwlandry merged 5 commits into from
Oct 2, 2026
Merged

Add nodes from dict #762

Show file tree
Hide file tree
Changes from 3 commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
45b1176
Add nodes from dict
nwlandry Sep 28, 2026
dcd0c5c
Fix failing test
nwlandry Sep 28, 2026
7fab2c7
fix broken doctests
nwlandry Sep 28, 2026
986cb74
added tests
nwlandry Sep 30, 2026
3a1690d
added additional logic and unit tests
nwlandry Oct 2, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 5 additions & 0 deletions tests/core/test_dihypergraph.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -81,6 +81,11 @@ def test_add_nodes_from(attr1, attr2, attr3):
assert H.nodes[1]["color"] == attr2["color"]
assert H.nodes[2]["color"] == attr3["color"]

H = xgi.DiHypergraph()
H.add_nodes_from({0: {"color": "black"}}, weight=1, color="grey")
assert H.nodes[0]["color"] == "black"
assert H.nodes[0]["weight"] == 1


def test_add_node_attr(diedgelist1):
H = xgi.DiHypergraph(diedgelist1)
Expand Down
5 changes: 5 additions & 0 deletions tests/core/test_hypergraph.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -143,6 +143,11 @@ def test_add_nodes_from(attr1, attr2, attr3):
assert H.nodes[1]["color"] == attr2["color"]
assert H.nodes[2]["color"] == attr3["color"]

H = xgi.Hypergraph()
H.add_nodes_from({0: {"color": "black"}}, weight=1, color="grey")
assert H.nodes[0]["color"] == "black"
assert H.nodes[0]["weight"] == 1


def test_remove_singleton_edges(edgelist1, edgelist2):
H1 = xgi.Hypergraph(edgelist1)
Expand Down
70 changes: 59 additions & 11 deletions xgi/core/dihypergraph.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -356,33 +356,81 @@ def add_nodes_from(self, nodes_for_adding, **attr):

Parameters
----------
nodes_for_adding : iterable
An iterable of nodes (list, dict, set, etc.).
nodes_for_adding : iterable or dict
An iterable of nodes (list, set, etc.).
OR
An iterable of (node, attribute dict) tuples.
OR
A dict mapping node IDs to attribute dictionaries.

Node attributes are updated using the attribute dict.
attr : keyword arguments, optional (default= no attributes)
attr : keyword arguments, optional (default=no attributes)
Update attributes for all nodes in nodes.
Node attributes specified in nodes as a tuple take
precedence over attributes specified via keyword arguments.
Node attributes specified for individual nodes take precedence
over attributes specified via keyword arguments.

See Also
--------
add_node
set_node_attributes

Examples
--------
>>> import xgi
>>> H = xgi.DiHypergraph()

Add nodes without attributes:

>>> H.add_nodes_from([1, 2, 3])

Add nodes with attributes shared by all nodes:

>>> H.add_nodes_from([1, 2, 3], color="red", weight=1)

Add nodes with individual attributes using (node, attribute dict)
tuples:

>>> H.add_nodes_from([
... (1, {"color": "red"}),
... (2, {"color": "blue"}),
... (3, {"color": "green"}),
... ])

Add nodes using a dict mapping node IDs to attribute dictionaries:

>>> H.add_nodes_from({
... 1: {"color": "red", "weight": 2},
... 2: {"color": "blue", "weight": 5},
... 3: {"color": "green"},
... })

Global attributes can also be combined with individual node
attributes. Individual node attributes take precedence:

>>> H.add_nodes_from(
... {
... 1: {"color": "red"},
... 2: {"color": "blue"},
... },
... weight=1,
... )
"""
if isinstance(nodes_for_adding, dict):
nodes_for_adding = nodes_for_adding.items()

for n in nodes_for_adding:
try:
newnode = n not in self._node
newdict = attr
except TypeError:
if isinstance(n, tuple) and len(n) == 2 and isinstance(n[1], dict):
n, ndict = n
newnode = n not in self._node
newdict = attr.copy()
newdict.update(ndict)
if newnode:
else:
newdict = attr

# Initialize new node if it doesn't already exist
if n not in self._node:
self._node[n] = {"in": set(), "out": set()}
self._node_attr[n] = self._node_attr_dict_factory()

self._node_attr[n].update(newdict)

def remove_node(self, n, strong=False, remove_empty=True):
Expand Down
70 changes: 59 additions & 11 deletions xgi/core/hypergraph.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -409,33 +409,81 @@ def add_nodes_from(self, nodes_for_adding, **attr):

Parameters
----------
nodes_for_adding : iterable
An iterable of nodes (list, dict, set, etc.).
nodes_for_adding : iterable or dict
An iterable of nodes (list, set, etc.).
OR
An iterable of (node, attribute dict) tuples.
OR
A dict mapping node IDs to attribute dictionaries.

Node attributes are updated using the attribute dict.
attr : keyword arguments, optional (default= no attributes)
attr : keyword arguments, optional (default=no attributes)
Update attributes for all nodes in nodes.
Node attributes specified in nodes as a tuple take
precedence over attributes specified via keyword arguments.
Node attributes specified for individual nodes take precedence
over attributes specified via keyword arguments.

See Also
--------
add_node
set_node_attributes

Examples
--------
>>> import xgi
>>> H = xgi.Hypergraph()

Add nodes without attributes:

>>> H.add_nodes_from([1, 2, 3])

Add nodes with attributes shared by all nodes:

>>> H.add_nodes_from([1, 2, 3], color="red", weight=1)

Add nodes with individual attributes using (node, attribute dict)
tuples:

>>> H.add_nodes_from([
... (1, {"color": "red"}),
... (2, {"color": "blue"}),
... (3, {"color": "green"}),
... ])

Add nodes using a dict mapping node IDs to attribute dictionaries:

>>> H.add_nodes_from({
... 1: {"color": "red", "weight": 2},
... 2: {"color": "blue", "weight": 5},
... 3: {"color": "green"},
... })

Global attributes can also be combined with individual node
attributes. Individual node attributes take precedence:

>>> H.add_nodes_from(
... {
... 1: {"color": "red"},
... 2: {"color": "blue"},
... },
... weight=1,
... )
"""
if isinstance(nodes_for_adding, dict):
nodes_for_adding = nodes_for_adding.items()

for n in nodes_for_adding:
try:
newnode = n not in self._node
newdict = attr
except TypeError:
if isinstance(n, tuple) and len(n) == 2 and isinstance(n[1], dict):
n, ndict = n
newnode = n not in self._node
newdict = attr.copy()
newdict.update(ndict)
if newnode:
else:
newdict = attr

# Initialize new node if it doesn't already exist
if n not in self._node:
self._node[n] = set()
self._node_attr[n] = self._node_attr_dict_factory()

self._node_attr[n].update(newdict)

def remove_node(self, n, strong=False, remove_empty=True):
Expand Down
Loading