def create_wordcloud(corpus, output, stopword_dict):
lex_dic = build_lex_dic(corpus, stopword_dict=stopword_dict)
total_words = get_total_words(lex_dic)
ordered_freq_list = build_freq_list(lex_dic, total_words)
fig = plt.figure(figsize=(10, 8), frameon=False)
ax = plt.Axes(fig, [0., 0., 1., 1.])
ax.set_axis_off()
fig.add_axes(ax)
wordcloud = WordCloud(width=1000, height=800, max_words=100, background_color='white',
relative_scaling=0.7, random_state=15, prefer_horizontal=0.5).generate_from_frequencies(
ordered_freq_list[0:100])
wordcloud.recolor(random_state=42, color_func=my_color_func)
ax.imshow(wordcloud)
fig.savefig(output, facecolor='white')
评论列表
文章目录