在Plotly气泡图中标记特定气泡

时间:2018-05-10 16:52:59

标签: python plot plotly

我无法弄清楚如何在plot.ly气泡图中标记特定的气泡。我想要某些"异常值"气泡将文本写入气泡内而不是通过悬停文本。

我们说我有这些数据:

import plotly.plotly as py
import plotly.graph_objs as go

trace0 = go.Scatter(
    x=[1, 2, 3, 4],
    y=[10, 11, 12, 13],
    mode='markers',
    marker=dict(
        size=[40, 60, 80, 100],
    )
)

data = [trace0]
py.iplot(data, filename='bubblechart-size')

我只想在与(1,10)和(4,13)对应的气泡上添加文字标记。此外,是否可以控制文本标记的位置?

1 个答案:

答案 0 :(得分:3)

您可以使用annotations实现此目的。

这允许您在图表上写下您想要的任何文本并将其引用到您的数据中。您还可以使用位置锚点或通过在x和y数据之上应用其他计算来控制文本的显示位置。例如:

x_data = [1, 2, 3, 4]
y_data = [10, 11, 12, 13]
z_data = [40, 60, 80, 100]

annotations = [
    dict(
        x=x, 
        y=y,
        text='' if 4 > x > 1 else z, # Some conditional to define outliers
        showarrow=False,
        xanchor='center',  # Position of text relative to x axis (left/right/center)
        yanchor='middle',  # Position of text relative to y axis (top/bottom/middle)
    ) for x, y, z in zip(x_data, y_data, z_data)
]

trace0 = go.Scatter(
    x=x_data,
    y=y_data,
    mode='markers',
    marker=dict(
        size=z_data,
    )
)

data = [trace0]
layout = go.Layout(annotations=annotations)

py.iplot(go.Figure(data=data, layout=layout), filename='bubblechart-size')

修改

如果使用袖扣,则可以稍微调整以上内容:

bubbles_to_annotate = df[(df['avg_pos'] < 2) | (df['avg_pos'] > 3)]  # Some conditional to define outliers
annotations = [
    dict(
        x=row['avg_pos'], 
        y=row['avg_neg'],
        text=row['subreddit'],
        showarrow=False,
        xanchor='center',  # Position of text relative to x axis (left/right/center)
        yanchor='middle',  # Position of text relative to y axis (top/bottom/middle)
    ) for _, row in bubbles_to_annotate.iterrows()
]

df.iplot(kind='bubble', x='avg_pos', y='avg_neg', size='counts', 
       text='subreddit', xTitle='Average Negative Sentiment', 
       yTitle='Average Positive Sentiment', annotations=annotations,
       filename='simple-bubble-chart')

由于需要条件参数,因此仍需要定义注释。然后通过annotations直接将其传递给袖扣。