如何在networkx图的绘图中绘制矩形?

时间:2019-02-26 13:44:39

标签: python matplotlib networkx

我有一个要绘制的图形,然后对其进行一些自定义。特别是,我想在一些节点组周围绘制框,并要编写文本。

到目前为止,我无法使其正常工作。我读到正确的方法是使用add_patches方法。

这是我的无效代码:

    import matplotlib.pyplot as plt   
    import networkx as nx
    from matplotlib.patches import Rectangle

    f = plt.figure(figsize=(16,10))

    G=nx.Graph()
    ndxs = [1,2,3,4]
    G.add_nodes_from(ndxs)
    G.add_weighted_edges_from( [(1,2,0), (1,3,1) , (1,4,-1) , (2,4,1) , (2,3,-1), (3,4,10) ] ) 
    nx.draw(G, nx.spring_layout(G, random_state=100))

    plt.gca().add_patch(Rectangle((50,100),40,30,linewidth=1,edgecolor='b',facecolor='none'))

我的问题是,最后一行似乎没有任何作用。

1 个答案:

答案 0 :(得分:1)

您的坐标在窗口之外。 如果运行plt.xlim()(或plt.ylim()),则会看到轴的范围接近[-1,1],而您试图在坐标[50,100]处设置矩形。

import matplotlib.pyplot as plt   
import networkx as nx
from matplotlib.patches import Rectangle

f,ax = plt.subplots(1,1, figsize=(8,5))

G=nx.Graph()
ndxs = [1,2,3,4]
G.add_nodes_from(ndxs)
G.add_weighted_edges_from( [(1,2,0), (1,3,1) , (1,4,-1) , (2,4,1) , (2,3,-1), (3,4,10) ] ) 
nx.draw(G)

ax.add_patch(Rectangle((0,0),0.1,0.1,linewidth=1,edgecolor='b',facecolor='none'))

enter image description here

我不熟悉networkx的工作原理,所以我不知道是否有一种方法可以正确计算所需矩形的坐标。一种方法是在轴坐标中绘制矩形(轴的左上角为0,0,右下角为1,1),而不是数据坐标:

import matplotlib.pyplot as plt   
import networkx as nx
from matplotlib.patches import Rectangle

f,ax = plt.subplots(1,1, figsize=(8,5))

G=nx.Graph()
ndxs = [1,2,3,4]
G.add_nodes_from(ndxs)
G.add_weighted_edges_from( [(1,2,0), (1,3,1) , (1,4,-1) , (2,4,1) , (2,3,-1), (3,4,10) ] ) 
nx.draw(G)

ax.add_patch(Rectangle((0.25,0.25),0.5,0.5,linewidth=1,edgecolor='b',facecolor='none', transform=ax.transAxes))

enter image description here