如何自定义matplotlib.patches.ArrowStyle？参数修改无效问题解决

首先得指出你代码里的核心问题：你用ArrowStyle("Test", ...)的时候，"Test"并不是matplotlib内置的箭头样式名称，也没有注册对应的自定义样式类，所以matplotlib会忽略这个无效的样式，自动 fallback 到默认箭头样式，这就是为什么你修改参数后看不到变化的原因。

下面给你两种解决方案，分别对应快速调整现有样式和完全自定义新样式的需求：

一、快速调整内置箭头样式的参数 #

如果只是想修改现有箭头的头部长度、宽度或者尾部宽度，直接用matplotlib内置的样式名称（比如"Fancy"、"Simple"、"->"等），并在参数里指定对应的属性即可。有两种写法：

写法1：在arrowprops里直接指定样式和参数 #

import matplotlib.pyplot as plt

plt.figure(figsize=(9,9))
ax = plt.subplot(111)

ax.annotate("", 
            xy=(0.2, 0.2), xycoords='data', 
            xytext=(0.8, 0.8), textcoords='data', 
            arrowprops=dict(
                # 用逗号分隔样式名称和参数
                arrowstyle="Fancy, head_length=0.6, head_width=0.4, tail_width=0.2",
                connectionstyle="arc3",
                color='crimson'
            )
)

plt.show()

写法2：创建ArrowStyle实例传入 #

import matplotlib.pyplot as plt
from matplotlib.patches import ArrowStyle

plt.figure(figsize=(9,9))
ax = plt.subplot(111)

# 先创建Fancy样式的实例，传入自定义参数
fancy_arrow = ArrowStyle("Fancy", head_length=0.4, head_width=0.3, tail_width=0.1)
ax.annotate("", 
            xy=(0.2, 0.6), xycoords='data', 
            xytext=(0.8, 0.6), textcoords='data', 
            arrowprops=dict(
                arrowstyle=fancy_arrow,
                connectionstyle="arc3",
                color='steelblue'
            )
)

plt.show()

注意：不同内置样式的参数名可能略有不同，比如"Simple"样式同样支持head_length、head_width、tail_width，你可以参考matplotlib文档确认对应样式的可用参数。

二、完全自定义新的ArrowStyle（高级需求） #

如果你需要设计独特的箭头形状，就要继承ArrowStyle._Base类，实现__call__方法来定义箭头的路径。下面是一个示例：

import matplotlib.pyplot as plt
from matplotlib.patches import ArrowStyle
from matplotlib.path import Path

# 定义自定义箭头样式类
class CustomArrow(ArrowStyle._Base):
    def __init__(self, head_length=0.4, head_width=0.4, tail_width=0.2):
        # 保存自定义参数
        self.head_len = head_length
        self.head_w = head_width
        self.tail_w = tail_width

    def __call__(self, path, mutation_size):
        # mutation_size是matplotlib自动传递的缩放因子，适配画布和箭头长度
        hl = self.head_len * mutation_size
        hw = self.head_w * mutation_size
        tw = self.tail_w * mutation_size

        # 定义箭头的顶点坐标（基于0-1的相对坐标）
        vertices = [
            (0.0, tw/2),          # 尾部右上
            (1.0 - hl, tw/2),     # 头部过渡右上
            (1.0 - hl, hw/2),     # 头部右上
            (1.0, 0.0),           # 箭头尖端
            (1.0 - hl, -hw/2),    # 头部右下
            (1.0 - hl, -tw/2),    # 头部过渡右下
            (0.0, -tw/2),         # 尾部右下
            (0.0, tw/2),          # 回到起点闭合路径
        ]

        # 定义路径的绘制指令
        codes = [
            Path.MOVETO,
            Path.LINETO,
            Path.LINETO,
            Path.LINETO,
            Path.LINETO,
            Path.LINETO,
            Path.LINETO,
            Path.CLOSEPOLY,
        ]

        # 返回定义好的路径
        return Path(vertices, codes)

# 注册自定义样式，这样就可以用名称调用了
ArrowStyle.register("MyCustomArrow", CustomArrow)

# 使用自定义样式
plt.figure(figsize=(9,9))
ax = plt.subplot(111)

# 方式1：用注册的名称加参数
ax.annotate("", 
            xy=(0.2, 0.2), xycoords='data', 
            xytext=(0.8, 0.8), textcoords='data', 
            arrowprops=dict(
                arrowstyle="MyCustomArrow, head_length=0.5, head_width=0.6, tail_width=0.3",
                connectionstyle="arc3",
                color='forestgreen'
            )
)

# 方式2：直接创建实例
custom_arrow = CustomArrow(head_length=0.3, head_width=0.4, tail_width=0.1)
ax.annotate("", 
            xy=(0.2, 0.6), xycoords='data', 
            xytext=(0.8, 0.6), textcoords='data', 
            arrowprops=dict(
                arrowstyle=custom_arrow,
                connectionstyle="arc3",
                color='darkviolet'
            )
)

plt.show()

在这个示例里，你可以通过修改vertices数组来调整箭头的形状，比如把头部改成圆角、添加额外的装饰等，完全自定义你的箭头样式。

内容的提问来源于stack exchange，提问作者Stücke