在Matplotlib中，有没有一种方法可以用圆或bar绘制一条线。在末端

Question

在matplotlib中，您可以使用axes.arrow（ https://matplotlib.org/3.5.0/api/_as_as_gen/matplotlib.axes.axes.axes.axes.arrow.html ）。

在末尾绘制带有圆或其他屁股的线的函数是否有类似的功能？我希望这是axes.arrow的关键字参数选项，但似乎不存在。

Answer 1

只是一个具有类和继承的想法，但我认为@Johanc链接更好：

from matplotlib.lines import Line2D
import matplotlib.pyplot as plt
import matplotlib.patches as pt


class Line2D(Line2D):
    def draw(self, rdr):
        super().draw(rdr)
        xy = self.get_xydata()
        start, end = xy[0], xy[-1]
        r = pt.Rectangle((start[0] - .05, start[1] - .05), .1, .1,
                         color=self.get_color(),
                         fill=None)
        plt.gca().add_patch(r)
        c = pt.Ellipse(end, .05, .05,
                       color=self.get_color(),
                       fill=True)
        plt.gca().add_patch(c)

fig = plt.figure()
ax = fig.add_subplot()
rg = [-.5, 1.5]
ax.set_xlim(rg)
ax.set_ylim(rg)

l = Line2D([0, 1], [0, 1], color="green")
ax.add_artist(l)

plt.show()

您可以获取其他参数（lineWidth等），以应用于您的补丁程序。