码迷,mamicode.com
首页 > 其他好文 > 详细

wordcloud简单上手

时间:2019-06-16 16:01:21      阅读:100      评论:0      收藏:0      [点我收藏+]

标签:img   理解   过程   width   inf   main   scipy   显示   __name__   

这只是一个十分简单的wordcloud的小尝试,只是在做的过程中遇到了一些问题,供大家参考。

import wordcloud
import matplotlib.pyplot as plt
import imageio


def wordcloud_test():
    # 需要转换成图片的文字文件
    f = open(D:\\prince.txt).read()
    f = f.lower()
    # 文字图片的形状
    m = imageio.imread(D:\\miku.jpeg)
    # 生成wordcloud对象
    w = wordcloud.WordCloud(background_color="black", width=800, height=660, margin=2, mask=m).generate(f)
    plt.imshow(w)
    plt.axis("off")
    plt.show()

    w.to_file(D:\\test.png)


if __name__ == __main__:
    wordcloud_test()

代码实现的效果就是将文件(prince.txt)中的信息使用wordcloud显示出来,形状采用的另一张图(miku.jpeg)的轮廓,最后返回结果(test.png)。

整个代码都是比较好理解的,唯独在导入模块使用imread方法的时候耽误了很长的时间,我参考的教程中imread的导入方法是"from scipy.misc import imread",

可是我在导入的时候总是提示没有这个方法。在网上找解决方法,有如下:

  1. 安装Pillow模块(我已安装)
  2. 安装Visual(安装后没有效果)

最后发现可以通过imageio模块使用imread方法,问题得以解决。如果有遇到类似问题的同学可以参考下。

附上效果图 

技术图片

wordcloud简单上手

标签:img   理解   过程   width   inf   main   scipy   显示   __name__   

原文地址:https://www.cnblogs.com/yuzhui/p/11031649.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!