如何为Matplotlib绘制的椭圆添加等高线式边缘标签？

要给手动绘制的椭圆边缘添加30、40、50这类数值标签，你可以通过计算标签在椭圆上的位置，再用Matplotlib的plt.annotate()函数实现。以下是修改后的完整代码：

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.patches import Ellipse

contours = [30, 40, 50]

v = np.array([15, 18])
colours = [
    (0.0, 0.26666666666666666, 0.10588235294117647, 1.0),
    (0.2917647058823529, 0.6886582083813917, 0.3827758554402153, 1.0),
    (0.8274509803921569, 0.9325490196078431, 0.8031372549019608, 1.0)
]

def ellipseFunction(x, v):
        return np.sqrt((x[0]/v[0])**2 + (x[1]/v[1])**2)

ellipses = [Ellipse((0, 0), width= 2*v[0]*contours[l], height=2*v[1]*contours[l], ec = colours[l], fc= "none") for l in range(len(contours))]

x = np.linspace(-2500, 2500, 100)
y = np.linspace(-2500, 2500, 100)
plt.figure()
plt.pcolormesh(x, y, ellipseFunction(np.meshgrid(x, y), v).T, cmap = 'plasma', zorder = -1, vmax = 150)
plt.gca().set_aspect(1./plt.gca().get_data_ratio())

for l in range(len(contours)):
    plt.gca().add_patch(ellipses[l])
    # 计算标签位置：椭圆右侧端点
    label_x = v[0] * contours[l]
    label_y = 0
    # 添加标注，微调位置避免和椭圆重叠
    plt.annotate(str(contours[l]), 
                xy=(label_x, label_y), 
                xytext=(label_x + 50, label_y),
                color=colours[l],
                fontweight='bold')

plt.show()

关键细节说明： #

• 标签位置选在椭圆右侧端点，你也可以根据需求调整到顶部(0, v[1]*contours[l])、左侧等其他位置。
• xytext参数用于偏移标签位置，避免和椭圆线条重叠，偏移量可根据图表范围自行调整。
• 标签颜色和对应椭圆的边缘颜色保持一致，保证视觉风格统一。

内容的提问来源于stack exchange，提问作者Puco4