先从库里面导入需要的模快
from wordcloud import WordCloud, STOPWORDS, ImageColorGenerator
import matplotlib.pyplot as plt
如果没有可以去终端里面下载,pip install 库名 回车即可。
将一个同级目录下的txt文件写入一些内容,我是爬取的B站上面一个视频的弹幕,下面是弹幕的部分截图,一共是一千条弹幕。
在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来进行查看!设置自己想设置的值,来观察不同的效果!
运行py文件即可显示类似如下的图片(此时显示的未加背景图片时运行出来的图片,后面有加了背景图时运行显示的图片,大家可以来看看区别!)
后面这个是使用背景图片
代码如下
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()
显示的图片如下