Блог

У меня есть некоторый код, который может обновлять атрибут графика:

 import networkx as nx

def update_nodes(graph):
    values = dict.fromkeys(graph.nodes, True)
    nx.set_node_attributes(graph, name='test_attribute', values=values)
    print(f"graph.nodes.data(): {graph.nodes.data()}")
    return graph

def loop(graph):
    graph.graph['test_attribute'] = False
    print(f"graph.nodes.data(): {graph.nodes.data()}")
    print(f"graph.graph['test_attribute']: {graph.graph['test_attribute']}")
    for node in range(0, len(graph.nodes)):
        print(f"graph.nodes[node]['test_attribute']: {graph.nodes[node]['test_attribute']}")
    return graph

graph = nx.erdos_renyi_graph(n = 3, p = 0.1, seed = 5)
for i in range(0, 2):
    graph = update_nodes(graph)
    graph = loop(graph)

 

Это не обновляет атрибут узла, а вместо этого применяет обновление к графу как к отдельному объекту. Есть ли какой-либо способ массового обновления одного атрибута всего набора узлов, не вставляя for node in (range(0, len(graph.nodes)): graph.nodes[node]['test_attribute'] = <new_value> где-нибудь?

Редактировать: я должен был уточнить, что причина, по которой я недоволен этой итерацией по содержимому графика, заключается в том, что это начальное значение необходимо сбросить для всех узлов перед запуском второго цикла после этого, что ставит под сомнение значение атрибута и следует различным логическим путям на основе этого. Поэтому я надеюсь избежать повторения всех узлов дважды, хотя этот первый цикл намного менее трудоемкий и, надеюсь, ненужный.

Если вы хотите установить для атрибута для всех узлов одинаковое значение, вы можете использовать graph.add_nodes_from(graph.nodes,attribute_name='attribute_value')

Это обновляет предоставленный атрибут (добавляет узел и атрибут, если они не существуют), но сохраняет любые другие атрибуты узла и ребра, которые у вас уже есть на графике. Смотрите пример ниже:

 import networkx as nx

graph = nx.erdos_renyi_graph(n = 3, p = 0.1, seed = 5)

# setup some initial attributes
graph.nodes[0]['attr1'] = 'foo'
graph.nodes[1]['attr1'] = 'bar'
graph.nodes[2]['attr1'] = 'baz'

# make some attribute we want to change
for node in graph.nodes:
    graph.nodes[node]['attr_to_change'] = 'orig_value'

# View initial attribute values
print(graph.nodes(data=True))
# [(0, {'attr1': 'foo', 'attr_to_change': 'orig_value'}),
#  (1, {'attr1': 'bar', 'attr_to_change': 'orig_value'}),
#  (2, {'attr1': 'baz', 'attr_to_change': 'orig_value'})]

# change the value of the attr_to_change attribute for all
graph.add_nodes_from(graph.nodes,attr_to_change='new_value')

# View updated attribute value
print(graph.nodes(data=True))
# [(0, {'attr1': 'foo', 'attr_to_change': 'new_value'}),
#  (1, {'attr1': 'bar', 'attr_to_change': 'new_value'}),
#  (2, {'attr1': 'baz', 'attr_to_change': 'new_value'})]