一些背景...
我是 python 和 networkx 的初学者...
我有一个包含多列的 csv 文件。
我提取了包含发件人和收件人地址的列,并将它们放入列表中,如下所示:
with open('file.csv,'r') as csv_file:
lines = csv_file.readlines()
sip = []
dip = []
for line in lines:
data = line.split(',')
sip.append(data[9])
dip.append(data[10])
nodes=list(set().union(sip,dip))
edges=list(set().union(list(zip(sip,dip))))
快速打印我的 G.nodes() 和 G.edges() 给我以下输出:
edges = [('AddressA','AddressB'),('AddressA','AddressC')]
nodes = ['AddressA','AddressB','AddressC']
其中边= [('senderaddress','receiveraddress')]
我的目标
我希望使用networkx来绘制发送者和接收者之间的连接。
我正在使用这个page作为引用。
这是我当前的代码:
import plotly.graph_objs as go
import networkx as nx
from plotly.subplots import make_subplots
#######CREATE NODES/EDGES#############
G=nx.Graph()
nodes=list(set().union(sip,dip))
edges=list(set().union(list(zip(sip,dip))))
G.add_nodes_from(nodes)
G.add_edges_from(edges)
pos = nx.get_node_attributes(G,'pos')
edge_trace = go.Scatter(
x=[],
y=[],
line=dict(width=0.5,color='#888'),
hoverinfo='none',
mode='lines')
for edge in G.edges():
x0, y0 = G.nodes[edge[0]]['pos']
x1, y1 = G.nodes[edge[1]]['pos']
edge_trace['x'] += tuple([x0, x1, None])
edge_trace['y'] += tuple([y0, y1, None])
node_trace = go.Scatter(
x=[],
y=[],
text=[],
mode='markers',
hoverinfo='text',
marker=dict(
showscale=True,
colorscale='YlGnBu',
reversescale=True,
color=[],
size=10,
colorbar=dict(
thickness=15,
title='Node Connections',
xanchor='left',
titleside='right'
),
line=dict(width=2)))
for node in G.nodes():
x, y = G.nodes[node]['pos']
node_trace['x'] += tuple([x])
node_trace['y'] += tuple([y])
# ########COLOR NODES#########
for node, adjacencies in enumerate(G.adjacency()):
node_trace['marker']['color']+=tuple([len(adjacencies[1])])
node_info = '# of connections: '+str(len(adjacencies[1]))
node_trace['text']+=tuple([node_info])
# ########CREATE GRAPH#########
fig = go.Figure(data=[edge_trace, node_trace],
layout=go.Layout(
title='<br>Network graph made with Python',
titlefont=dict(size=16),
showlegend=False,
hovermode='closest',
margin=dict(b=20,l=5,r=5,t=40),
annotations=[ dict(
text="Python code: <a href='https://plot.ly/ipython-notebooks/network-graphs/'> https://plot.ly/ipython-notebooks/network-graphs/</a>",
showarrow=False,
xref="paper", yref="paper",
x=0.005, y=-0.002 ) ],
xaxis=dict(showgrid=False, zeroline=False, showticklabels=False),
yaxis=dict(showgrid=False, zeroline=False, showticklabels=False)))
fig.show()
但是,我最终遇到了这个错误:
x0, y0 = G.nodes[edge[0]]['pos']
KeyError: 'pos'
打印(pos)显示它是空的:
{}
我不知道错误出在哪里。我相信自从我用节点和边填充图表以来它可能已经开始了。但我不知道如何纠正它......
最佳答案
我修改了发布的代码的某些部分。 从相关输出中直接给出节点和边。
import plotly.graph_objs as go
import networkx as nx
from plotly.subplots import make_subplots
x = []
y = []
G=nx.Graph()
#from question edges = [('AddressA','AddressB'),('AddressA','AddressC')] nodes = ['AddressA','AddressB','AddressC']
print ("nodes =", nodes)
print ("edges =", edges)
G.add_nodes_from(nodes, word1 = 'word2')
pos = nx.get_node_attributes(G,'word1')
G.add_edges_from(edges)
print("pos = ", pos)
这将打印 pos
注1:
如果我在 pos = nx.get_node_attributes(G,'word1') 之前 G.add_edges_from ; 那么 pos = []
如果我在 G.add_edges_from 之前分配 pos = nx.get_node_attributes(G,'word1') ,那么 pos 将正确打印
我得到的输出如下:
nodes = ['AddressA', 'AddressB', 'AddressC']
edges = [('AddressA', 'AddressB'), ('AddressA', 'AddressC')]
pos = {'AddressA': 'word2', 'AddressB': 'word2', 'AddressC': 'word2'}
关于python - 带有 csv 输入和绘图输出的 networkx 图,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/58526661/