使用plotly绘制气泡图

debugcn 发表于 Dev

用户名

我正在使用plotly在气泡图中绘制一些数据。我已经编写了以下代码：（基于Plotly的示例）

trace0 = go.Scatter(
    x=output['mean'],
    y=output['count'],
    text=output.index,
    mode='markers',
    marker=dict(
        color=['rgb(93, 164, 214)', 'rgb(255, 144, 14)',  'rgb(44, 160, 101)', 'rgb(255, 65, 54)'],
        size=output['total'],
    )
)

data = [trace0]
plot_url = py.plot_mpl(trace0, filename='mpl-7d-bubble')

我正在尝试output['mean']在x轴和output['count']y上进行绘制。我想在output['total'].运行时绘制大小，收到此错误：

Traceback (most recent call last):
  File "..data.py", line 45, in <module>
    plot_url = py.plot_mpl(data, filename='mpl-7d-bubble')
  File "...anaconda/lib/python3.5/site-packages/plotly/plotly/plotly.py", line 307, in plot_mpl
    fig = tools.mpl_to_plotly(fig, resize=resize, strip_style=strip_style)
  File ".../anaconda/lib/python3.5/site-packages/plotly/tools.py", line 507, in mpl_to_plotly
    matplotlylib.Exporter(renderer).run(fig)
  File ".../anaconda/lib/python3.5/site-packages/plotly/matplotlylib/mplexporter/exporter.py", line 45, in run
    fig.savefig(io.BytesIO(), format='png', dpi=fig.dpi)
AttributeError: 'list' object has no attribute 'savefig'

编辑：output熊猫数据框的内容：

             count     location        mean        total
StartLocation                                             
?                  6            ?    9.527778    57.166667
bathroom          14     bathroom   20.409524   285.733333
bedroom           61      bedroom   96.837432  5907.083333
bedroom2          22     bedroom2  165.262121  3635.766667
hallway            6      hallway    0.394444     2.366667
kitchen           79      kitchen    8.646624   683.083333
living room       56  living room   93.855655  5255.916667
outside            6      outside  325.991667  1955.950000
setup              1        setup    0.050000     0.050000
study             18        study   18.099074   325.783333
toilet            51       toilet    7.198693   367.133333

马克西米利安·彼得斯（Maximilian Peters）

有几个小问题使您无法获得情节。

阴暗地期望data成为一个plotly.graph_objs.Data对象
```
data = go.Data([trace0])
```
您的尺寸标记太大，您需要将其最小化到合理的水平
```
size=[i / 100 for i in output['total']]
```

上图的完整代码。

import pandas as pd
import plotly.plotly as py
import plotly.graph_objs as go
import io

txt = """room count location mean total
? 6 ? 9.527778 57.166667
bathroom 14 bathroom 20.409524 285.733333
bedroom 61 bedroom 96.837432 5907.083333
bedroom2 22 bedroom2 165.262121 3635.766667
hallway 6 hallway 0.394444 2.366667
kitchen 79 kitchen 8.646624 683.083333
livingroom 56 livingroom 93.855655 5255.916667
outside 6 outside 325.991667 1955.950000
setup 1 setup 0.050000 0.050000
study 18 study 18.099074 325.783333
toilet 51 toilet 7.198693 367.133333"""

t = io.StringIO(txt)
output = pd.read_csv(t, sep=' ')

trace0 = go.Scatter(
    x=output['mean'],
    y=output['count'],
    text=output.room,
    mode='markers',
    marker=dict(
        color=['rgb(93, 164, 214)', 'rgb(255, 144, 14)',  'rgb(44, 160, 101)', 'rgb(255, 65, 54)'],
        size=[i / 100 for i in output['total']],
    )
)

data = go.Data([trace0])
plot_url = py.plot(data, filename='mpl-7d-bubble')

本文收集自互联网，转载请注明来源。

如有侵权，请联系[email protected] 删除。