python - 如何在粗线上获得非圆形箭头？

Question

我正在使用 networkx 在 Python 中绘制有向图。我使用不同宽度的边缘来突出重量。不幸的是，箭头是圆形的，看起来很奇怪。我想为看起来像细线箭头的缩放版本的粗线绘制非圆形箭头。

import networkx as nx
import matplotlib.pyplot as plt
import numpy

G=nx.DiGraph()
stake = [
    [0, 1, 1, 3],
    [3, 0, 0, 1],
    [0, 0, 0, 1],
    [1, 0, 3, 0]
    ]
maxS = max(max(stake))
stake1d = numpy.concatenate(stake)
minS = min(stake1d[numpy.nonzero(stake1d)])
minLineWeight = 1
maxLineWeight = 10

for x in range(len(stake)):
    for y in range(x):
        if(stake[x][y] > 0):
            weight = (stake[x][y] - minS) / (maxS - minS) * (maxLineWeight - minLineWeight) + minLineWeight
            G.add_edge(x, y, weight=weight, color='r')

for x in range(len(stake)):
    for y in [i for i in range(len(stake))][-(len(stake)-x):]:
        if(stake[x][y] > 0):
            weight = (stake[x][y] - minS) / (maxS - minS) * (maxLineWeight - minLineWeight) + minLineWeight
            G.add_edge(x, y, weight=weight, color='b')

weights=list(nx.get_edge_attributes(G,'weight').values())
colors=list(nx.get_edge_attributes(G,'color').values())
pos = nx.shell_layout(G)
nx.draw(
    G,
    pos=pos,
    width=weights,
    edge_color=colors,
    with_labels=True,
    arrows=True,
    connectionstyle='arc3',
    arrowstyle='->'
)  
plt.show()

Answer 1

我有这个完全相同的问题并一起破解了一个解决方法:

1. 将 nx.draw 调用替换为两个单独的方法调用，分别绘制节点和边。这很有用，因为 networkx 边绘制方法 draw_networkx_edges 返回一个 matplotlib 补丁列表(特别是 FancyArrowPatch 对象)，然后您可以在渲染之前对其进行操作。

2. 遍历箭头补丁并更改 FancyArrowPatch mutation_scale ，以及 joinstyle和 capstyle构成每个箭头的线段。

解决方法

从您的示例代码开始，将对 nx.draw 的调用替换为以下内容。

# First, draw the nodes themselves
nodes = nx.draw_networkx_nodes(
    G,
    pos=pos,
    linewidths=1
)

# Draw node labels
node_label_handles = nx.draw_networkx_labels(
    G, 
    pos=pos
);

# Draw the edges and store the returned FancyArrowPatch list
arrows = nx.draw_networkx_edges(
    G,
    pos=pos,
    arrows=True,
    width=weights,
    edge_color=colors,
    arrowstyle='-|>'  # I personally think this style scales better
)

for a, w in zip(arrows, weights):
    
    # mutation_scale affects only the arrowhead size, not the arrow tail.
    # The constants here are arbitrary; you may want/need to change them
    a.set_mutation_scale(20 + w)
    
    # Sharpen arrowheads by creating a mitered joint between arrowhead 
    # line segments, instead of the default joinstyle='round'
    a.set_joinstyle('miter')
    
    # Prevent each arrow tail from jutting forward and through the arrowhead,
    # which happens with the default capstyle='projecting'
    a.set_capstyle('butt')

我的结果

版本

pandas=1.3.2, numpy=1.21.2, matplotlib=3.4.3