2

我正在尝试从熊猫数据框创建一个交互式绘图图。

但是，我无法正确显示图例。

这是一个工作示例：

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
import plotly.plotly as py

# sign into the plotly api
py.sign_in("***********", "***********")

# create some random dataframes
dates = pd.date_range('1/1/2000', periods=8)
df1 = pd.DataFrame(np.random.randn(8, 1), index=dates, columns=['A'])
df2 = pd.DataFrame(np.random.randn(8, 1), index=dates, columns=['B'])
df1.index.name = 'date'
df2.index.name = 'date'

现在我尝试使用 plotly 绘制数据框。

fig, ax = plt.subplots(1,1)
    
df1.plot(y='A', ax=ax)
df2.plot(y='B', ax=ax)
    
py.iplot_mpl(fig, filename='random')

注意没有图例

剧情 - 没有传说

编辑：

根据以下建议，我添加了一个update字典。尽管这确实显示了图例，但它却弄乱了情节本身：

fig, ax = plt.subplots(1,1)
    
df1.plot(y='A', ax=ax)
df2.plot(y='B', ax=ax)
    
update = dict(
    layout=dict(
        annotations=[dict(text=' ')],  # rm erroneous 'A', 'B', ... annotations
        showlegend=True                # show legend 
    )
)
py.iplot_mpl(fig, update=update, filename='random')

在此处输入图像描述

编辑2：

从dict中删除annotations条目layout会导致绘图正确显示，但图例不是y列名，而是x列名，数据框的索引名

fig, ax = plt.subplots(1,1)
    
df1.plot(y='A', ax=ax)
df2.plot(y='B', ax=ax)
    
update = dict(
    layout=dict(
        showlegend=True                # show legend 
    )
)
py.iplot_mpl(fig, update=update, filename='random')

这导致以下情节：

在此处输入图像描述

编辑3：

我找到了一种覆盖图例文本的方法，但它似乎有点笨拙。鉴于我已经指定了要绘制的数据框列：

df1.plot(y='A', ax=ax)

我原以为这y='A'会导致'A'被用作图例标签。

似乎情况并非如此，虽然可以使用索引标签覆盖，如下所示，但感觉不对。

有没有更好的方法来实现这个结果？

update = dict(
    layout=dict(
        showlegend=True,
    ),
    data=[
        dict(name='A'),
        dict(name='B'),
    ]
)
py.iplot_mpl(fig, update=update, filename='random')

在此处输入图像描述

4

1 回答 1

3

图例不能很好地从 matplotlib 转换为 plotly。

幸运的是，在 matplotlib 图中添加一个情节图例是直截了当的：

update = dict(
    layout=dict(
        showlegend=True  # show legend 
    )
)
py.iplot_mpl(fig, update=update)

在此处查看完整的工作ipython 笔记本。

有关详细信息，请参阅 plotly用户指南。

于 2015-04-29T14:52:16.123 回答