python - plotly : How to annotate multiple lines in Plotly Express?

Question

我想创建一个折线图，其中多条线共享相同的 x 轴和 y 轴。 幸运的是，最新版本的 Plotly 似乎允许多个输入作为 y。 例子: px.line(df,x,[y1,y2]) 工作正常。

不幸的是，我想添加文本作为数据点注释，但这并不以同样的方式工作。

我的目标是创建一个包含多条线的线图，每个数据点都用与相应线相同颜色的 y_value 进行注释。

感谢您的帮助!

Answer 1

也许不是最优雅的方法，但下面的完整代码片段将产生下图。代码片段的一些核心部分是:

方法:

for i, d in enumerate(fig.data):
    for j, a in enumerate(d.x):
        fig.add_annotation(x=a, y = d.y[j], text = str(d.y[j])[:5],
                          showarrow = False,
                          yshift = 10,
                          font=dict(color=d.line.color, size=12))

图 1:

如果您想关注其他color cycles对于您的注释，只需包含:

colors = px.colors.qualitative.Alphabet

并替换:

font=dict(color=d.line.color, size=12)

与:

font=dict(color=colors[i], size=12)

并得到:

图 2:

如果您可以使用它，我很乐意详细介绍所有细节。

完整代码:

# imports
import pandas as pd
import plotly.express as px

# data
df = px.data.stocks().tail(10)
df = df.drop(['AMZN', 'AAPL'], axis = 1)

df.set_index('date', inplace = True)
colors = px.colors.qualitative.Alphabet

fig = px.line(df, x = df.index, y = df.columns)

for i, d in enumerate(fig.data):
    for j, a in enumerate(d.x):
        fig.add_annotation(x=a, y = d.y[j], text = str(d.y[j])[:5],
                           showarrow = False,
                           yshift = 10,
                           font=dict(color=d.line.color, size=12)
#                            font=dict(color=colors[i], size=12)
                          )


fig.show()