NetworkX：如何将节点坐标分配为属性？

Question

在像这样的简单图表中：

import networkx as nx
import matplotlib.pyplot as plt

G = nx.Graph()
G.add_edge('0','1')
G.add_edge('1','2')
G.add_edge('2','0')
G.add_edge('0','3')
G.add_edge('1','4')
G.add_edge('5','0')

pos={'0':(1,0),'1':(1,1),'2':(2,3),'3':(3,2),'4':(0.76,1.80),'5':(0,2)} #node:(x,y)
nx.draw(G,pos=pos,with_labels=True)
plt.show()

如果我尝试为每个节点分配一个包含节点 ID 及其 (x,y) 坐标的属性列表，如下所示：

for i,n in enumerate(G.nodes()):
    G.nodes()[i]['weight']=[G.nodes()[i],pos[n]] #List of attributes

我收到以下错误：

Traceback (most recent call last):

  File "<ipython-input-47-0f9ca94eeefd>", line 2, in <module>
    G.nodes()[i]['weight']=[G.nodes()[i],pos[n]] 

TypeError: 'str' object does not support item assignment

这里有什么问题？

Answer 1

经过一些研究，我发现答案在 nx.set_node_attributes()。

当然可以将节点位置分配为属性：

pos={'0':(1,0),'1':(1,1),'2':(2,3),'3':(3,2),'4':(0.76,1.80),'5':(0,2)}    
nx.set_node_attributes(G, pos, 'coord')

这导致

In[1]: G.nodes(data=True)
Out[1]:
[('1', {'coord': (1, 1)}), #each node has its own position
 ('0', {'coord': (1, 0)}),
 ('3', {'coord': (3, 2)}),
 ('2', {'coord': (2, 3)}),
 ('5', {'coord': (0, 2)}),
 ('4', {'coord': (0.76, 1.8)})]

并且还可以使用专用字典（在本例中为 test）附加多个属性，这些属性不必与 G 中的节点具有相同数量的元素（例如, 可以有节点 without attributes):

test={'0':55,'1':43,'2':17,'3':86,'4':2} #node '5' is missing
nx.set_node_attributes(G, 'test', test)

这导致

In[2]: G.nodes(data=True)
Out[2]:
[('1', {'coord': (1, 1), 'test': 43}),
 ('0', {'coord': (1, 0), 'test': 55}),
 ('3', {'coord': (3, 2), 'test': 86}),
 ('2', {'coord': (2, 3), 'test': 17}),
 ('5', {'coord': (0, 2)}),
 ('4', {'coord': (0.76, 1.8), 'test': 2})]

我推测使用 nx.set_edge_attributes().

也可以对图的边进行同样的操作

NetworkX：如何将节点坐标分配为属性？

NetworkX: how to assign the node coordinates as attribute?

python

graph

typeerror

networkx