如何在Matplotlib非数值轴绘图中获取数据点坐标？

要解决这个问题，核心是要明白：matplotlib在处理字符串类型的x轴时，会把每个字符串标签映射成从0开始的整数坐标（比如第一个标签对应0，第二个对应1，以此类推）。你可以通过以下步骤获取实际绘图坐标并添加偏移标注：

具体步骤与代码示例 #

假设你的DataFrame和绘图代码如下：

import pandas as pd
import matplotlib.pyplot as plt

# 示例数据
df = pd.DataFrame({
    'x_label': ['Jan', 'Feb', 'Mar', 'Apr', 'May'],
    'y_value': [12, 28, 18, 35, 29]
})

# 绘制折线图，获取折线对象
fig, ax = plt.subplots(figsize=(8,4))
line, = ax.plot(df['x_label'], df['y_value'], marker='o')

接下来按以下操作处理：

1. 获取实际的绘图坐标

   • 折线的y值可以直接通过line.get_data()[1]获取
   • x轴的实际绘图坐标（整数位置）用ax.get_xticks()获取，这会返回和字符串标签一一对应的数值（比如[0,1,2,3,4]）

   y_data = line.get_data()[1]
   x_pos = ax.get_xticks()  # 得到每个x标签对应的实际绘图x坐标
   

2. 遍历线段，计算中点并添加偏移标注
   循环每一对相邻的点，计算线段中点，再根据斜率调整标注的偏移位置：

   for i in range(len(x_pos) - 1):
       # 计算线段中点坐标
       mid_x = (x_pos[i] + x_pos[i+1]) / 2
       mid_y = (y_data[i] + y_data[i+1]) / 2
       
       # 计算线段斜率
       slope = (y_data[i+1] - y_data[i]) / (x_pos[i+1] - x_pos[i])
       
       # 根据斜率设置标注偏移（可根据需求调整偏移量）
       # 斜率为正，标注向上偏移；斜率为负，向下偏移，同时加水平偏移避免遮挡
       offset_y = 1.5 if slope > 0 else -1.5
       offset_x = 0.05
       
       # 添加标注，这里示例标注线段的变化值
       change = y_data[i+1] - y_data[i]
       ax.text(mid_x + offset_x, mid_y + offset_y, 
               f'{change:+}', fontsize=10, ha='center')
   

3. 调整图表显示
   最后可以调整x轴标签的显示，避免重叠：

   plt.xticks(rotation=45)
   plt.tight_layout()
   plt.show()
   

关键说明 #

• ax.get_xticks()是核心：它直接给出了字符串x标签对应的实际绘图坐标，这是line.get_data()返回字符串所没有的信息
• 偏移量可以根据你的图表大小、标注内容自行调整，确保标注既不遮挡折线，又清晰可见

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