我正在创建一个以节点为图像的图形,
#http://matplotlib.sourceforge.net/users/image_tutorial.html中的图片
我想创建一个圆形布局,将节点zero
置于中心。egdelist为[(0,1),(0,2),(0,3),(0,4), (0,5)]
import matplotlib.pyplot as plt
import matplotlib.image as mpimg
import numpy as np
import networkx as nx
img=mpimg.imread('stinkbug.png')
G=nx.complete_graph(6)
G.node[0]['image']=img
G.node[1]['image']=img
G.node[2]['image']=img
G.node[3]['image']=img
G.node[4]['image']=img
G.node[5]['image']=img
print(G.nodes())
G.add_edge(0,1)
G.add_edge(0,2)
G.add_edge(0,3)
G.add_edge(0,4)
G.add_edge(0,5)
print(G.edges())
nx.draw_circular(G)
但是,在输出中我发现了额外的边缘(附有快照),是否有办法去除这些额外的边缘?我只希望这些条件Egdelist是[(0,1),(0,2),(0,3),(0,4),(0,5)]。此外,原始图像未显示在节点中。
有什么建议吗?
答案 0 :(得分:1)
所以这里确实有两个问题。第一个是为什么图形的边比您想要的多。发生这种情况是因为您使用nx.complete_graph(6)
来初始化图形-这在6个节点上创建了完整的图形。您应该初始化一个空图,添加带有图像元数据的节点,然后添加边缘。
要绘制节点作为您的图像,我从this discussion找到并稍加修改了代码。您可以自定义一些内容,例如图像大小。结果是:
希望这会有所帮助!
import matplotlib.pyplot as plt
import matplotlib.image as mpimg
import numpy as np
import networkx as nx
img=mpimg.imread('/Users/johanneswachs/Downloads/stink.jpeg')
G=nx.Graph()
G.add_node(0,image= img)
G.add_node(1,image= img)
G.add_node(2,image= img)
G.add_node(3,image= img)
G.add_node(4,image= img)
G.add_node(5,image= img)
print(G.nodes())
G.add_edge(0,1)
G.add_edge(0,2)
G.add_edge(0,3)
G.add_edge(0,4)
G.add_edge(0,5)
print(G.edges())
pos=nx.circular_layout(G)
fig=plt.figure(figsize=(5,5))
ax=plt.subplot(111)
ax.set_aspect('equal')
nx.draw_networkx_edges(G,pos,ax=ax)
plt.xlim(-1.5,1.5)
plt.ylim(-1.5,1.5)
trans=ax.transData.transform
trans2=fig.transFigure.inverted().transform
piesize=0.2 # this is the image size
p2=piesize/2.0
for n in G:
xx,yy=trans(pos[n]) # figure coordinates
xa,ya=trans2((xx,yy)) # axes coordinates
a = plt.axes([xa-p2,ya-p2, piesize, piesize])
a.set_aspect('equal')
a.imshow(G.node[n]['image'])
a.axis('off')
ax.axis('off')
plt.show()