Plotly：如何使用散点图显示多索引数据帧？

Question

我正在使用多索引数据框绘制折线图。使用Matplotlib绘制图形时，我可以看到正确的结果，但使用Plotly散点图绘制时，数据框显示错误的输出-为什么？

import pandas as pd
data = pd.DataFrame([
        ('Q1','Blue',100),
        ('Q1','Green',300),
        ('Q2','Blue',200),
        ('Q2','Green',350),
        ('Q3','Blue',300),
        ('Q3','Green',400),
        ('Q4','Blue',400),
        ('Q4','Green',450),
    ], 
    columns=['quarter', 'company', 'value']
)
data = data.set_index(['quarter', 'company']).value

data.unstack().plot(kind='bar', stacked=True)

上面的代码绘制了正确的图表。下面的代码也会生成正确的结果。

fig = go.Figure()
fig.add_trace(go.Scatter(x=sample.index, y=sample['Blue'], 
                         #mode='lines+markers', name='',
                         #line=dict(color=colors_list[1],width=2,) ,
                          
                        )
)

但是我不知道如何同时在散点图中绘制蓝色和绿色？

例子：

fig = go.Figure()
fig.add_trace(go.Scatter(x=sample.index, y=sample, 
                         #mode='lines+markers', name='',
                         #line=dict(color=colors_list[1],width=2,) ,
                          
                        )
)

谁能帮我把蓝色和绿色画在一起？

Answer 1

正如维斯特兰在他的回答中所指出的那样，你可以使用Plotly Express来简单地完成它。

但是，如果不想使用Plotly Express，则必须向地物添加绘图（迹线），并按如下所示配置地物键。

import plotly.graph_objects as go

fig = go.Figure()
fig.add_trace(go.Scatter(x=sample.index, y=sample['Green'], mode='markers', name = 'Green', marker_color = 'Green'))
fig.add_trace(go.Scatter(x=sample.index, y=sample['Blue'], mode='markers', name = 'Blue', marker_color = 'Blue'))
fig.update_layout(title="The Title", xaxis_title="Quarter", yaxis_title="Value", legend_title="Company")

给予

Answer 2

您的问题有点不清楚，但我假设您在这里的主要目标是显示一个季度数组中的值，其中值分为两组[“蓝色”、“红色”]。（我不太明白为什么您要求提供一个绘图仪散点图，但显示一个matplotlib条形图…）。无论如何，如果我是对的，那么您的用例非常适合plotly.express，特别是px.scatter，下面的代码片段将使用它生成下图。

import pandas as pd
import plotly.graph_objects as go
import plotly.express as px

data = pd.DataFrame([
        ('Q1','Blue',100),
        ('Q1','Green',300),
        ('Q2','Blue',200),
        ('Q2','Green',350),
        ('Q3','Blue',300),
        ('Q3','Green',400),
        ('Q4','Blue',400),
        ('Q4','Green',450),
    ], 
    columns=['quarter', 'company', 'value']
)

fig = px.scatter(data, x = 'quarter', y = 'value',
                 color = 'company',
                 color_discrete_sequence=['Blue', 'Green'])
fig.show()