你的第一个WorldCloud

Contents


一、 前言

我们经常看到网上一些漂亮的词云图,Python里就有一个这样的包来完成这样的功能,他就是 WorldCloud包。

二、 你的第一个 WorldCloud

  1. 安装:pip install wordcloud
  2. 代码
     from wordcloud import WordCloud
    
     f = open('Alice.txt','r').read()
     wordcloud = WordCloud(background_color="white",width=1000, height=860, margin=2).generate(f)
    
     # width,height,margin 可以设置图片属性
    
     # generate 可以对全部文本进行自动分词,但是他对中文支持不好
     #wordcloud = WordCloud(font_path = r'D:\Fonts\simkai.ttf').generate(f)
     # 你可以通过 font_path 参数来设置字体集
    
     #background_color 参数为设置背景颜色,默认颜色为黑色
    
     import matplotlib.pyplot as plt
     plt.imshow(wordcloud)
     plt.axis("off")
     plt.show()
    
     # 保存图片,但是在第三模块的例子中 图片大小将会按照 mask 保存
     wordcloud.to_file('Alice1.png')
    
    
     plt.imshow(wordcloud, interpolation='bilinear')
     plt.axis("off")
     wordcloud.to_file('Alice2.png')
    

三、 小结

上面是一个最简单的例子,我们可以使用它来练练手,当然,你还可以使用 WorldCloud包来制作各种图形和颜色的词云图。后面我们会介绍结合 jieba 分词包来使用,生成漂亮的中文词云。

References

[1] wordcloud · PyPI

Further Reading

[1] jieba · PyPI


转载请注明:yezuolin的博客 » 点击阅读原文