时间戳

首页 > 解决方案 > Matplotlib - 用三角形框注释图

python - Matplotlib - 用三角形框注释图

问题描述

我想用一个尖尖的三角形框来注释我的情节的左上角，里面有一个字母，如下图所示。我只能做一个方形盒子，但我想要一个三角形盒子。

plt.annotate("A",
             xy = (0.05, 0.92),
             xycoords = "axes fraction",
             bbox = dict(boxstyle = "square", fc = "red"))

左上角带有红色三角形的插图图，里面有字母 A

标签： pythonmatplotlib

解决方案

从这里，您可以绘制任何您想要的任意形状。

这不完全是你想要的，但可能会让你通过。


import numpy as np
import matplotlib.pyplot as plt
import matplotlib.path as mpath
import matplotlib.patches as mpatches
from matplotlib.collections import PatchCollection


def label(xy, text):
    y = xy[1] - 0.15  # shift y-value for label so that it's below the artist
    plt.text(xy[0], y, text, ha="center", family='sans-serif', size=14)


fig, ax = plt.subplots()

patches = []

# add a path patch
Path = mpath.Path
path_data = [
    (Path.MOVETO, [0.00, 0.00]),
    (Path.LINETO, [0.0, 1.0]),
    (Path.LINETO, [1.0, 1.0]),
    (Path.CLOSEPOLY, [0.0, 1.0])
    ]
codes, verts = zip(*path_data)
path = mpath.Path(verts, codes)
patch = mpatches.PathPatch(path)
patches.append(patch)

colors = np.linspace(0, 1, len(patches))
collection = PatchCollection(patches, alpha=0.3)
collection.set_array(np.array(colors))
ax.add_collection(collection)

plt.subplots_adjust(left=0, right=1, bottom=0, top=1)
plt.axis('equal')
plt.axis('off')

plt.show()

推荐阅读