Matplotlib均匀间隔标记点绘制重叠问题求助

问题描述 #

我是编程新手，在信号处理场景中遇到问题：基于Matplotlib编写文本框控件代码，希望通过文本框输入标记数量，在图表上绘制均匀间隔的标记点。目前使用numpy linspace函数，从0到len(x_data)-1生成指定数量的索引，但出现标记点重叠的情况——输入10个标记却只显示5个重叠的点。

初始代码 #

def process_points(event):
    ax2.cla()
    
    number_of_points = int(text2_box1.text)
    marker_indices = np.linspace(0, len(x_data)-1, number_of_points, dtype=int)
    x_markers = x_data[marker_indices]
    y_markers = y_data[marker_indices]
    
    ax2.plot(x_data, y_data, color='blue')
    ax2.set_xlabel('Frequency (Hz)')
    ax2.set_ylabel('Amplitude')
    ax2.set_title('Data between lines')
    
    for x, y in zip(x_markers, y_markers):
        ax2.plot(x, y, '.', color='crimson')
    
    plt.show()

初始效果：标记点重叠，多个点集中在同一位置

问题原因 #

问题出在np.linspace(0, len(x_data)-1, number_of_points, dtype=int)这一行：

• 用dtype=int强制转换linspace生成的浮点索引时，会直接截断小数部分，导致多个浮点索引被转换成同一个整数索引，最终选中的数据点重复，表现为标记点重叠。

解决后的代码 #

def process_points(event):
     ax2.cla()

     number_of_points = int(text2_box1.text)
     x_markers = np.linspace(0, x_data[-1], number_of_points)

     ax2.semilogy(x_data, y_data, color='blue')
     ax2.set_xlabel('Frequency (Hz)')
     ax2.set_ylabel('Amplitude')
     ax2.set_title('Data between lines')

     y_markers = np.interp(x_markers, x_data, y_data)  
     ax2.scatter(x_markers, y_markers, color='crimson')  

     ax2.grid(True, which='both', axis='both')

     plt.show()

更新后效果：标记点在曲线上均匀分布，无重叠

优化说明 #

• 直接在x轴数据范围内生成均匀间隔的x_markers，而非通过索引取点，避免了整数索引截断的问题。
• 使用np.interp函数根据原始x、y数据插值计算对应x_markers的y值，确保标记点精准落在曲线上。
• 改用scatter绘制标记点，比循环调用plot更高效，显示效果也更清晰。
• 添加网格并改用semilogy绘制曲线，更适配信号处理中频率-幅度的可视化需求。

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