Python 将瀑布图转换为 plotly

Question

我正在使用 Python 中的 waterfall_chart 包来创建瀑布图。该包主要在后端使用 matplotlib，所以我尝试使用 tls.mpl_to_plotly(mpl_fig) 函数将 matplotlib 图转换为 plotly 。但是转换的时候会报错。有没有一种方法可以将 waterfall_chart 转换为 plotly 或者是否有一种简单的方法可以直接在 plotly 中创建图表？我之前看到一些关于类似 chart 的讨论在 plotly 中，但它涉及图表编号的相当手动编码。

您可以使用以下代码重新创建图表。

import waterfall_chart
import matplotlib.pyplot as plt
import plotly.tools as tls

a = ['sales','returns','credit fees','rebates','late charges','shipping']
b = [10,-30,-7.5,-25,95,-7]
mpl_fig = plt.figure()
waterfall_chart.plot(a, b)

plt.show()

waterfall chart

但是当我尝试使用 mpl_to_plotly() 转换为 plotly 时，出现错误:

plotly_fig = tls.mpl_to_plotly(mpl_fig)

ValueError: min() arg is an empty sequence

waterfall_chart 包的详细信息可以在这里找到:https://github.com/chrispaulca/waterfall/blob/master/waterfall_chart.py

Answer 1

我的回答地址

[...] or is there an easy way to create the chart directly in plotly?

---

对于较新版本的 plotly，您可以使用 plotly.graph_objs.Waterfall。

下面是您的数据样本的基本示例，其设置在离线 Jupyter Notebook 中使用 iplot:

剧情:

代码:

# imports
import plotly
from plotly.offline import download_plotlyjs, init_notebook_mode, plot, iplot
from IPython.core.display import display, HTML
import plotly.figure_factory as ff
import plotly.graph_objs as go
import pandas as pd
import numpy as np

# setup
display(HTML("<style>.container { width:35% !important; } .widget-select > select {background-color: gainsboro;}</style>"))
init_notebook_mode(connected=True)
np.random.seed(1)

import plotly.offline as py
import plotly.graph_objs as go
py.init_notebook_mode(connected = False)

# your values
a = ['sales','returns','credit fees','rebates','late charges','shipping']
b = [10,-30,-7.5,-25,95,-7]

# waterfall trace
trace = go.Waterfall(
     x = a, 
    textposition = "outside", 
    text = [str(elem) for elem in b], 
    y = b, 
    connector = {"line":{"color":"rgb(63, 63, 63)"}}, 
)

layout = go.Layout(
        title = "Waterfall chart, plotly version 3.9.0", 
        showlegend = True
)

iplot(go.Figure([trace], layout))

---

检查你的版本:

import plotly
plotly.__version__

使用 cmd 控制台更新您的版本:

pip install plotly --upgrade