如何在 Python Plotly 中在悬停时突出显示一组中的所有值？

在数据可视化中，悬停时突出显示数据的小技巧能够让我们更好地理解数据。在 Plotly 中，我们可以使用“hovertemplate”参数来实现这一功能。

Plotly 简介

Plotly 是一个 Python 数据可视化工具，可以创建交互式、幻灯片和网页版的图表。

在本篇文章中，我们将使用 Plotly 包中的 Scatter 类来创建散点图，并使用“hovertemplate”参数来突出显示该组中的所有值。

首先，我们需要安装 Plotly 包。可使用以下命令：

pip install plotly

除此之外，我们还需要导入以下模块：

import plotly.express as px
from plotly.subplots import make_subplots
import plotly.graph_objects as go

准备工作

首先，我们需要创建一组数据。在这里，我们将创建由三个点组成的 2D 数据点。

import numpy as np

x = np.array([1, 2, 3])
y = np.array([2, 4, 3])

接下来，我们使用 Plotly 来创建该组数据的散点图，并添加“hovertemplate”参数。

# 创建散点图
fig = go.Figure(data=[go.Scatter(
    x=x,
    y=y,
    mode='markers',
    hovertemplate='<b>X: %{x}</b><br>' +
                  '<b>Y: %{y}</b><br>' +
                  '<b>Group:</b> All values'
)])

fig.show()

在运行此代码后，将显示一个散点图，当悬停在任何一个数据点上时，都会在右下角显示所有值。

添加组别

为了能够将相同组别的数据点突出显示，我们需要将它们分类。在这里，我们将设置一个与数据点相同长度的标记数组，这个标记数组包含每个数据点所属的组编号。

# 将数据点分类
groups = np.array([1, 2, 1])

接下来，我们将使用 make_subplots 方法来创建数据点和组别的两个子图。

fig = make_subplots(rows=1, cols=2)

scatter = go.Scatter(
    x=x,
    y=y,
    mode='markers',
    hovertemplate='<b>X: %{x}</b><br>' +
                  '<b>Y: %{y}</b><br>' +
                  '<b>Group:</b> %{text}',
    text=groups  # 添加组别信息
)

# 添加数据点子图
fig.add_trace(scatter, row=1, col=1)

# 添加组别子图
fig.add_trace(
    go.Scatter(x=groups, y=[1, 1, 1], mode='markers',
               marker=dict(size=10, color=groups)),
    row=1, col=2
)

fig.show()

此时，我们将看到两个子图。左边的子图是数据点的散点图，右边的子图是组别的颜色图。

突出显示组别

最后，我们需要调整“hovertemplate”参数，使用“text”参数将其设置为组别。

对于每个数据点，我们还需使用一个布尔值数组来判断它是否属于要突出显示的组别，并使用“marker.color”来改变数据点的颜色。

# 突出显示组别
fig = go.Figure()

for i in range(len(x)):
    # 判断数据点是否属于突出显示的组别
    if groups[i] == 1:
        marker_color = 'red'
        text = 'Group 1'
    else:
        marker_color = 'blue'
        text = 'Group 2'

    # 将突出显示组别的数据点以红色显示
    fig.add_trace(go.Scatter(
        x=[x[i]],
        y=[y[i]],
        mode='markers',
        text=text,
        marker=dict(color=marker_color, size=10),
        hovertemplate='<b>X: %{x}</b><br>' +
                      '<b>Y: %{y}</b><br>' +
                      '<b>Group:</b> %{text}'
    ))

fig.show()

此时，我们将看到只有那些属于突出显示组别的散点以红色显示。