如何在 dash web 应用程序上显示 wordcloud 图像

Question

我需要在我的 dash 应用程序上渲染一个 wordcloud。根据这个线程https://community.plot.ly/t/solved-is-it-possible-to-make-a-wordcloud-in-dash/4565，dash 中没有 wordcloud 内置组件。一种解决方法是使用WordCloud 模块将wordcloud 生成为图像，并使用dash_html_components.Img 在布局上显示。

我是 Dash 的新手。不知道如何渲染图像。每次生成 wordcloud 时都需要将 wordcloud 保存为临时图像吗？

如果有 Dash 方面的专业知识的人可以提供帮助，我将不胜感激。

代码如下：

import dash
import dash_core_components as dcc
import dash_html_components as html

print(dcc.__version__) # 0.6.0 or above is required

external_stylesheets = ['https://codepen.io/chriddyp/pen/bWLwgP.css']

app = dash.Dash(__name__, external_stylesheets=external_stylesheets)

dfm = pd.DataFrame({'word': ['apple', 'pear', 'orange'], 'freq': [1,3,9]})

app.layout = html.Div([
    html.Img(id = 'image_wc')
])

# function to make wordcoud from word frequency dataframe
def plot_wordcloud (data):
        d = {}
        for a, x in data.values:
            d[a] = x
        wc = WordCloud(background_color='black',
                    width=1800,
                    height=1400).generate_from_frequencies(frequencies=d)   
        return (wc)

@app.callback(dash.dependencies.Output('image_wc', 'img'))
def make_image ():
    img = plot_wordcloud (data = dfm)
    return (img)


if __name__ == '__main__':
    app.run_server(debug=True)

Answer 1

下面是一个工作示例。它使用pip-installable wordcloud 库，并通过BytesIO 传递结果图像的base-64 编码PNG 表示，因此您无需每次都转储所有创建的PNG 文件。

我已经让它自己触发，这样当你加载 Dash 应用程序时它就会运行，尽管你可以让它通过一个按钮或类似的东西动态地工作。

import dash
import dash.dependencies as dd
import dash_core_components as dcc
import dash_html_components as html

from io import BytesIO

import pandas as pd
from wordcloud import WordCloud
import base64

# external_stylesheets = ['https://codepen.io/chriddyp/pen/bWLwgP.css']

app = dash.Dash(__name__) #, external_stylesheets=external_stylesheets)

dfm = pd.DataFrame({'word': ['apple', 'pear', 'orange'], 'freq': [1,3,9]})

app.layout = html.Div([
    html.Img(id="image_wc"),
])

def plot_wordcloud(data):
    d = {a: x for a, x in data.values}
    wc = WordCloud(background_color='black', width=480, height=360)
    wc.fit_words(d)
    return wc.to_image()

@app.callback(dd.Output('image_wc', 'src'), [dd.Input('image_wc', 'id')])
def make_image(b):
    img = BytesIO()
    plot_wordcloud(data=dfm).save(img, format='PNG')
    return 'data:image/png;base64,{}'.format(base64.b64encode(img.getvalue()).decode())

if __name__ == '__main__':
    app.run_server(debug=True)