如何在Matplotlib中通过点击图例隐藏绘图中的折线？

要实现点击图例就能隐藏对应折线，关键得保存每条折线的对象引用，再给图例绑定点击事件来切换线条的可见性。下面是针对你代码的修改方案：

第一步：修正原代码问题并保存线条实例 #

你的原代码用plt.plot一次性画出了所有折线，但没单独保存每条线的实例，也没加图例。先修改out函数，逐个绘制折线并保存，同时添加图例标签：

import numpy as np
import matplotlib.pyplot as plt

# 修正detect函数里的笔误和缺失逻辑：np.dort→np.sort，补充diff定义和距离矩阵计算
def detect(df_abs,a,b,c):
    X = df_abs.iloc[:, a:b].to_numpy()
    l = X.shape[0]
    D = np.zeros((l, l))
    # 示例：计算欧氏距离矩阵（你可以换成自己的距离计算逻辑）
    for i in range(l):
        for j in range(l):
            D[i,j] = np.linalg.norm(X[i] - X[j])
    D = np.maximum(D, D.transpose())
    d = D.sum(0) / l
    diff = np.diff(np.sort(d))  # 补充diff的定义
    t = np.sort(d)[np.where(diff > c * np.median(diff))[0] + 1].min()
    return df_abs.iloc[np.where(d >= t)].index

def out(df_abs, a, b, c):
    o = detect(df_abs, a, b, c)
    plt.figure(figsize=(20, 10))
    
    # 逐个绘制折线，保存每个线条对象
    lines = []
    cols = df_abs.columns[1:]  # 取第1列之后的列名作为图例标签
    for col in cols:
        line, = plt.plot(df_abs[col], label=col, color='blue')
        lines.append(line)
    
    # 创建图例
    leg = plt.legend()
    
    # 定义图例点击的回调函数
    def on_legend_click(event):
        # 找到点击的图例条目对应的原始折线
        leg_line = event.artist
        orig_line = next(line for line in lines if line.get_label() == leg_line.get_label())
        # 切换折线可见性
        visible = not orig_line.get_visible()
        orig_line.set_visible(visible)
        # 调整图例条目的透明度，直观显示状态
        leg_line.set_alpha(1.0 if visible else 0.2)
        plt.draw()
    
    # 给每个图例条目开启拾取功能，绑定点击事件
    for leg_line in leg.get_lines():
        leg_line.set_picker(True)
    plt.gcf().canvas.mpl_connect('pick_event', on_legend_click)
    
    plt.show()

核心逻辑解释 #

• 保存线条对象：用lines列表存下每个plt.plot返回的线条实例，这样能通过图例标签匹配到对应的折线。
• 绑定点击事件：通过mpl_connect把pick_event和回调函数关联，点击图例时触发切换：修改折线的visible属性，同时调整图例条目的透明度来区分显示/隐藏状态。
• 原代码修复：你原来的detect函数里有np.dort（应该是np.sort）、diff未定义的问题，我补充了距离矩阵的计算示例，你可以根据自己的实际需求替换这部分逻辑。

使用方式 #

调用out函数时传入你的DataFrame和参数a,b,c，运行后点击图例里的任意标签，对应的折线就会隐藏或重新显示。

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