生成词云

先从库里面导入需要的模快

from wordcloud import WordCloud, STOPWORDS, ImageColorGenerator
import matplotlib.pyplot as plt

如果没有可以去终端里面下载,pip install 库名  回车即可。

将一个同级目录下的txt文件写入一些内容,我是爬取的B站上面一个视频的弹幕,下面是弹幕的部分截图,一共是一千条弹幕。

生成词云_第1张图片

在python中输入如下代码,其中background_image是你自己找的背景图片,不过图片要求是二值图(非黑即白),在最后显示出来的图片会在黑色区域显示字,白色区域仍是空白。在括号内输入你图片的路径,前面的r是为了防止转义的!这个font_path可以使用电脑里面自带的字体,不过也是写入路径。

from wordcloud import WordCloud, STOPWORDS, ImageColorGenerator
import matplotlib.pyplot as plt

f = open('弹幕.txt', 'r', encoding='utf-8').read()  ##生成词云的文档

# background_image = plt.imread(r'C:\Users\Administrator\Desktop\图片.jpg')  #背景图

wordcloud = WordCloud(
    background_color='pink',  # 背景颜色,默认黑色
    font_path=r'C:\Windows\Fonts\微软雅黑\msyh.ttf',  # 设置字体
    # mask=background_image,  # 背景图
    width=1000,  # 宽度和高度的像素值
    height=800,
    # min_font_size=6,  # 默认为4
    # max_font_size=90,
    margin=10).generate(f)
plt.imshow(wordcloud)
plt.axis('off')  # 坐标轴关闭
plt.show()

剩下还有一些值也是可以设置的,想试的人可以central+鼠标左键点击图中的WordCloud来进行查看!设置自己想设置的值,来观察不同的效果!

生成词云_第2张图片

 

运行py文件即可显示类似如下的图片(此时显示的未加背景图片时运行出来的图片,后面有加了背景图时运行显示的图片,大家可以来看看区别!)

生成词云_第3张图片

后面这个是使用背景图片

生成词云_第4张图片

代码如下

from wordcloud import WordCloud, STOPWORDS, ImageColorGenerator
import matplotlib.pyplot as plt

f = open('弹幕.txt', 'r', encoding='utf-8').read()  ##生成词云的文档

background_image = plt.imread(r'C:\Users\Administrator\Desktop\图片.jpg')  #背景图

wordcloud = WordCloud(
    background_color='pink',  # 背景颜色,默认黑色
    font_path=r'C:\Windows\Fonts\微软雅黑\msyh.ttf',  # 设置字体
    mask=background_image,  # 背景图
    width=1000,  # 宽度和高度的像素值
    height=800,
    # min_font_size=6,  # 默认为4
    # max_font_size=90,
    margin=10).generate(f)
plt.imshow(wordcloud)
plt.axis('off')  # 坐标轴关闭
plt.show()

显示的图片如下

生成词云_第5张图片

你可能感兴趣的:(生成词云)