主页 / 问题 / 1589160
Accepted
Angelina
Asked:2024-08-01 20:43:17 +0000 UTC2024-08-01 20:43:17 +0000 UTC

如何在图表中靠近反向边缘的地方指定两个不同颜色的箭头？网络x，Python

  • 772

我试着把反面的肋骨做成不同的颜色。但我不能 =( 现在如果你看一下图表，边 AL 和 LA 的颜色相同：

图形

如何使一个方向的反向边缘（例如，蓝色）和相反方向的橙色？而且筋条的厚度还取决于筋条的重量。我想保留直肋骨，它们的颜色和线条也有所不同。

使用networkx、matplotlib.pyplot、matplotlib.patches、python 我的代码：

#Визуальное разделение прямых и обратных ребер
connectionstyle = []
edge_colors=[]
 
for v,u in G.edges():
    if G1.has_edge(v,u):
        connectionstyle.append('arc3, rad=0')
        edge_colors.append('#000000')
    else:
        connectionstyle.append('arc3, rad=0.4')
        edge_colors.append('#4682B4')

#Выводим основной график
Arrow = mpl.ArrowStyle.CurveB(head_length=0.3, head_width=.4) #стиль стрелки
pos = nx.shell_layout(G) #расположение нодов
 
fig, ax = plt.subplots(figsize=(8,7))
edges = nx.draw_networkx_edges(G, pos, node_size=200, arrowsize=15, arrowstyle=Arrow, width=width)
 
#Стиль отображения узлов
for p in pos:
    if pos[p][0] < 0:
        pos[p][0] -= 0.02
        plt.text(pos[p][0],pos[p][1],s=p, bbox=dict(facecolor='#87CEFA', edgecolor='None', alpha=0.5),
             horizontalalignment='right', fontsize=11)
    else:
        pos[p][0] += 0.02
        plt.text(pos[p][0],pos[p][1],s=p, bbox=dict(facecolor='#87CEFA', edgecolor='None', alpha=0.5),
             horizontalalignment='left', fontsize=11)
 
#Отображение весов ребер
edge_labels = nx.get_edge_attributes(G2,'weight')
 
#Задаем параметры отображения линий, т.е. прямых и обратных ребер:
M = G.number_of_edges()
for i in range(M):
    edges[i].set_connectionstyle(connectionstyle[i])
    edges[i].set_color(edge_colors[i])
    edges[i].set_alpha(0.7)
 
 
#Выводим текстом инф о прямых ребрах:
xmin, xmax = ax.get_xlim()
ymin, ymax = ax.get_ylim()
 
s = ' \n'.join([v+'-->'+u+'  '+str(w['weight']) for v,u,w in G1.edges.data()])
plt.text(1.4,-0.2, s='Прямые ребра \n\n'+s, 
         bbox=dict(facecolor='#87CEFA', edgecolor='None', alpha=0.8),
         fontsize=11, horizontalalignment='center', color='white')
 
plt.tight_layout()
plt.axis('off')
plt.savefig('net2.svg')
plt.show()
python

1 个回答

  1. Best Answer

    该代码不允许我重现该问题。

    据我所理解：

    • G - 这些是正向和反向的边缘；
    • G1- 这些是未使用的肋骨；
    • G2- 还有别的东西。

    那些。G可能已使用以下参数进行初始化：

    G = nx.DiGraph()
G.add_edges_from([('A', 'L'), ('L', 'A'),
                  ('A', 'C'),
                  ('A', 'F'),
                  ('A', 'I'), ('I', 'A'),
                  ('C', 'L'),
                  ('I', 'C'), ('C', 'I'),
                  ('I', 'J'),
                  ('J', 'K')
                  # какие-то ещё...
                  ]
                 )

    我认为最明显的解决方案：

    connectionstyle = []
edge_colors=[]
check_reverse = []

for v, u in G.edges():

    if G1.has_edge(v, u):
        connectionstyle.append('arc3, rad=0')
        edge_colors.append('#000000')

    elif (v, u) not in check_reverse:
        connectionstyle.append('arc3, rad=0.4')
        edge_colors.append('#4682B4')

        if G.has_edge(u, v):
            check_reverse.append((u, v))

    elif (v, u) in check_reverse:
        connectionstyle.append('arc3, rad=0.4')
        edge_colors.append('#FFA500')
        
else:
    del check_reverse

# Далее подстраивайте свой код...

    要点是，在代码的这一部分中，您创建了两个相同长度的列表。
    而且所有的肋骨G原本都被赋予了蓝色。

    • 1

相关问题