如何在Matplotlib中保留自动刻度格式化并标注坐标轴上的点？

好的，我来帮你实现这两种需求，下面分别给出详细的代码和说明，确保注释能跟着视图缩放、刻度变化同步更新：

这种方式会在横轴下方对应点的x坐标位置，或者纵轴左侧对应点的y坐标位置添加自定义注释，当点在视图内时显示，超出视图时自动隐藏。

我们可以扩展原来的PointMarker类，加入注释文本的逻辑，在每次视图更新时同步调整注释的位置和可见性：

import numpy as np
import matplotlib.pyplot as plt

class PointMarker():
    def __init__(self, ax, point, annot_text, **kwargs):
        self.ax = ax
        self.point = point
        self.annot_text = annot_text
        if "line" in kwargs:
            self.c = kwargs.get("line").get_color()
        else:
            self.c = kwargs.get("color", "b")
        self.ls = kwargs.get("linestyle", ':')
        # 初始化辅助线
        self.vline, = self.ax.plot([],[],color=self.c,linestyle=self.ls)
        self.hline, = self.ax.plot([],[],color=self.c,linestyle=self.ls)
        # 初始化横轴下方的注释（可改为纵轴左侧）
        self.x_annot = self.ax.text(0, 0, self.annot_text["x"], 
                                    ha='center', va='top', color=self.c,
                                    transform=self.ax.get_xaxis_transform())
        self.y_annot = self.ax.text(0, 0, self.annot_text["y"], 
                                    ha='right', va='center', color=self.c,
                                    transform=self.ax.get_yaxis_transform())
        self.draw()

    def draw(self):
        xlim = self.ax.get_xlim()
        ylim = self.ax.get_ylim()
        x, y = self.point

        # 更新辅助线
        self.vline.set_data([x, x], [ylim[0], y])
        self.hline.set_data([xlim[0], x], [y, y])

        # 判断点是否在视图内
        in_view = xlim[0] <= x <= xlim[1] and ylim[0] <= y <= ylim[1]
        
        # 更新横轴注释：位置在x对应点，y轴下限下方
        if in_view:
            self.x_annot.set_x(x)
            self.x_annot.set_y(0)  # y=0是轴的底部（相对于x轴变换）
            self.x_annot.set_visible(True)
        else:
            self.x_annot.set_visible(False)
        
        # 更新纵轴注释：位置在y对应点，x轴左限左侧
        if in_view:
            self.y_annot.set_y(y)
            self.y_annot.set_x(0)  # x=0是轴的左侧（相对于y轴变换）
            self.y_annot.set_visible(True)
        else:
            self.y_annot.set_visible(False)

class PointMarkers():
    pointmarkers = []
    def add(self, ax, point, annot_text, **kwargs ):
        pm = PointMarker(ax, point, annot_text, **kwargs)
        self.pointmarkers.append(pm)
    def update(self, event=None):
        for pm in self.pointmarkers:
            pm.draw()

# 主程序
x = np.arange(1,17)
y = np.log(x)
ax = plt.subplot(111)
line = plt.plot(x,y)

# 注册标记和注释：自定义x轴和y轴的注释文本
p = PointMarkers()
p.add(ax, [x[5], y[5]], 
      annot_text={"x": "my_point_X", "y": "my_point_Y"},
      line=line[0])

# 绑定视图更新事件
cid = plt.gcf().canvas.mpl_connect("draw_event", p.update)

plt.grid(True)
plt.show()

关键说明： #

• 使用transform=self.ax.get_xaxis_transform()确保注释始终对齐x轴，不受数据坐标缩放影响
• 通过判断点是否在当前轴的xlim和ylim范围内，控制注释的显示/隐藏
• 你可以调整注释的位置参数（比如va='bottom'或者ha='left'）来改变注释的对齐方式

这种方式会把和目标点坐标重合的自动刻度标签替换成自定义文本，其他刻度保留Matplotlib的自动格式化效果，缩放时会自动检测新的刻度并替换：

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.ticker import FuncFormatter

class PointMarker():
    def __init__(self, ax, point, **kwargs):
        self.ax = ax
        self.point = point
        if "line" in kwargs:
            self.c = kwargs.get("line").get_color()
        else:
            self.c = kwargs.get("color", "b")
        self.ls=kwargs.get("linestyle", ':')
        self.vline, = self.ax.plot([],[],color=self.c,linestyle=self.ls)
        self.hline, = self.ax.plot([],[],color=self.c,linestyle=self.ls)
        self.draw()
    def draw(self):
        xmin = self.ax.get_xlim()[0]
        ymin = self.ax.get_ylim()[0]
        self.vline.set_data([self.point[0], self.point[0]], [ymin,self.point[1]])
        self.hline.set_data([xmin, self.point[0]], [self.point[1], self.point[1]])

class PointMarkers():
    pointmarkers = []
    def add(self,ax, point, **kwargs ):
        pm = PointMarker(ax, point, **kwargs)
        self.pointmarkers.append(pm)
    def update(self, event=None):
        for pm in self.pointmarkers:
            pm.draw()

# 自定义刻度格式化函数：替换目标点的刻度标签
def custom_x_formatter(x, pos, target_x, custom_text):
    # 考虑浮点精度，判断刻度值是否接近目标点x坐标
    if np.isclose(x, target_x, atol=1e-3):
        return custom_text
    # 其他刻度保留原自动格式化
    return f"{x:.0f}"  # 可以根据需求调整格式，比如保留小数

def custom_y_formatter(y, pos, target_y, custom_text):
    if np.isclose(y, target_y, atol=1e-3):
        return custom_text
    return f"{y:.2f}"

# 主程序
x = np.arange(1,17)
y = np.log(x)
ax = plt.subplot(111)
line = plt.plot(x,y)

# 目标点
target_x = x[5]
target_y = y[5]

# 注册标记
p = PointMarkers()
p.add(ax, [target_x, target_y], line=line[0])

# 应用自定义格式化器到x轴和y轴
ax.xaxis.set_major_formatter(FuncFormatter(lambda x, pos: custom_x_formatter(x, pos, target_x, "my_point_X")))
ax.yaxis.set_major_formatter(FuncFormatter(lambda y, pos: custom_y_formatter(y, pos, target_y, "my_point_Y")))

# 绑定视图更新事件（确保缩放后刻度重新计算时格式化器生效）
cid = plt.gcf().canvas.mpl_connect("draw_event", p.update)

plt.grid(True)
plt.show()

关键说明： #

• 使用FuncFormatter包装自定义逻辑，既保留Matplotlib的自动刻度生成，又能替换目标刻度
• np.isclose用于处理浮点精度问题，避免因为坐标是浮点数导致匹配失败
• 你可以调整格式化函数里的返回格式（比如f"{x:.0f}"）来匹配原自动刻度的显示风格

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