python - 文本位置未显示在绘图上

Question

我正在尝试使用plotly绘制我的神经网络训练和测试集的准确性。 我还想添加一个带有文本的标记，该文本说明每个值的最大值何时出现，而且还显示一个文本说明该值是什么。我尝试做类似this example中的事情.

这是我的 mcve:

import plotly.graph_objects as go

data = {
    'test acc':  [1, 2, 3, 4, 5, 6, 7, 9, 10],
    'train acc': [3, 5, 5, 6, 7, 8, 9, 10, 8]
}

fig = go.Figure()
color_train = 'rgb(255, 0, 0)'
color_test = 'rgb(0, 255, 0)'
assert len(data["train acc"]) == len(data["test acc"])
x = list(range(len(data["train acc"])))
fig.add_trace(go.Scatter(x=x,
                         y=data["train acc"],
                         mode='lines',
                         name='train acc',
                         line_color=color_train))
fig.add_trace(go.Scatter(x=x,
                         y=data["test acc"],
                         mode='lines',
                         name='test acc',
                         line_color=color_test))
# Max points
train_max = max(data["train acc"])
test_max = max(data["test acc"])
# ATTENTION! this will only give you first occurrence
train_max_index = data["train acc"].index(train_max)
test_max_index = data["test acc"].index(test_max)

fig.add_trace(go.Scatter(x=[train_max_index],
                         y=[train_max],
                         mode='markers',
                         name='max value train',
                         text=['{}%'.format(int(train_max * 100))],
                         textposition="top center",
                         marker_color=color_train))
fig.add_trace(go.Scatter(x=[test_max_index],
                         y=[test_max],
                         mode='markers',
                         name='max value test',
                         text=['{}%'.format(int(test_max*100))],
                         textposition="top center",
                         marker_color=color_test))

fig.update_layout(title='Train vs Test accuracy',
                  xaxis_title='epochs',
                  yaxis_title='accuracy (%)'
                  )
fig.show()

但是，我的输出火如下:

如您所见，该值并未像我找到的示例中那样显示。 我怎样才能让它出现？

Answer 1

如果您只想突出显示几个特定值，请使用 add_annotation()。在您的情况下，只需找到您想要聚焦的 X 的最大和最小 Y 值即可。由于您缺乏数据样本，我将如何使用通用数据样本来做到这一点:

plotly :

代码:

import plotly.graph_objects as go
import plotly.io as pio

pio.renderers.default='browser'

fig = go.Figure()

xVars1=[0, 1, 2, 3, 4, 5, 6, 7, 8]
yVars1=[0, 1, 3, 2, 4, 3, 4, 6, 5]


xVars2=[0, 1, 2, 3, 4, 5, 6, 7, 8]
yVars2=[0, 4, 5, 1, 2, 2, 3, 4, 2]

fig.add_trace(go.Scatter(
    x=xVars1,
    y=yVars1
))


fig.add_trace(go.Scatter(
    x=xVars2,
    y=yVars2
))

fig.add_annotation(
            x=yVars1.index(max(yVars1)),
            y=max(yVars1),
            text="yVars1 max")
fig.add_annotation(
            x=yVars2.index(max(yVars2)),
            y=max(yVars2),
            text="yVars2 max")
fig.update_annotations(dict(
            xref="x",
            yref="y",
            showarrow=True,
            arrowhead=7,
            ax=0,
            ay=-40
))

fig.update_layout(showlegend=False)

fig.show()