绘图：将图例添加到子绘图

Question

我正在使用python和plotly，并且尝试向每个子图添加一个图例。哪个不起作用...

当我运行以下代码时，

import plotly
import numpy as np

fig = plotly.subplots.make_subplots(rows=2, cols=1)

y = np.arange(0,10,1)

fig.add_trace(go.Scatter(y=y,name="name1"), row=1,col=1)
fig.add_trace(go.Scatter(y=y**2,name="name2"), row=1,col=1)


fig.add_trace(go.Scatter(y=y,name="name3"), row=2,col=1)
fig.add_trace(go.Scatter(y=y**2,name="name4"), row=2,col=1)

fig.show()

这就是我得到的：

但是我想要这样的东西：

有什么主意吗？

谢谢！

Answer 1

当我查看它时，每个子图的图例似乎与this information相对应。您在那里建议使用注释作为替代方法，因此我将它们添加到了您的代码中。

import plotly.graph_objects as go
from plotly.subplots import make_subplots
import numpy as np

fig = plotly.subplots.make_subplots(rows=2, cols=1)

y = np.arange(0,10,1)

fig.add_trace(go.Scatter(y=y,name="name1"), row=1,col=1)
fig.add_trace(go.Scatter(y=y**2,name="name2"), row=1,col=1)


fig.add_trace(go.Scatter(y=y,name="name3"), row=2,col=1)
fig.add_trace(go.Scatter(y=y**2,name="name4"), row=2,col=1)

for row in [1,2]:
    if row == 1:
        fig.add_annotation(dict(x=1.0, y=0.7, xref="paper", yref="paper", 
                                text='name %d' %row, showarrow=False))
        fig.add_annotation(dict(x=1.0, y=1.0, xref="paper", yref="paper", 
                                text='name %d' %(row+1), showarrow=False))
    else:
        fig.add_annotation(dict(x=1.0, y=0.05, xref="paper", yref="paper", 
                                text='name %d' %(row+1), showarrow=False))
        fig.add_annotation(dict(x=1.0, y=0.4, xref="paper", yref="paper", 
                                text='name %d' %(row+2), showarrow=False))

fig.show()