使用Networkx绘制格子和图形

Question

我想用Networkx绘制正方形点阵。我做了这样的事情：

import matplotlib.pyplot as plt
import numpy as np
import networkx as nx
L=4

G = nx.Graph()

pos={}
for i in np.arange(L*L):
   pos[i] = (i/L,i%L)

nx.draw_networkx_nodes(G,pos,node_size=50,node_color='k')    

plt.show()

然而输出只是一个空白数字。我该如何解决这个问题？

另外，我想用箭头水平和垂直地连接点。从(i,j) to (i+1,j)开始的箭头方向应该取决于我已经拥有的矩阵A的i，j元素的符号。怎么做？

Answer 1

这是nx.grid_2d_graph的显式图构造函数：

G = nx.grid_2d_graph(L,L)
nx.draw(G,node_size=2000)
plt.show()

我们可以将此无向图修改为符合边缘标准的有向图。举个例子，我删除了远离原点的边缘。您可以根据自己的需要进行调整：

G2 = nx.DiGraph(G)
for edge in G2.edges():
    if edge != tuple(sorted(edge)):
        G2.remove_edge(*edge)

nx.draw_spectral(G2,node_size=600,node_color='w')