配合上次的词云
颜色可控调整
用colormap : string or matplotlib colormap, default=”viridis” //给每个单词随机分配颜色,若指定color_func,则忽略该方法。
不会用color_func方法,所以就找到一个这种方法
代码
import matplotlib.pyplot as plt
import cv2
import jieba
from wordcloud import WordCloud
from matplotlib import colors
f=open(r'F:\python\python课程设计\弹幕8000.txt','r',encoding='utf-8')
text=f.read()
cut_text = "".join(jieba.cut(text))
print(cut_text)
color_list=['#FF0000','#a41a1a']#建立颜色数组
colormap=colors.ListedColormap(color_list)#调用
color_mask = cv2.imread("11.jpg")
word_cloud=WordCloud(
font_path="FZLTCHJW.TTF",
background_color=None,
mode="RGBA",
prefer_horizontal=1,
mask=color_mask,
height=200,
width=200,
scale=1,
colormap=colormap,#设置颜色
margin=5
)
word_cloud1=word_cloud.generate(cut_text)
word_cloud1.to_file('chengpin.png')
plt.imshow(word_cloud, interpolation='bilinear')
plt.axis('off')
plt.show()
这是在上一篇的基础上添加了3行代码
成品:
也是参考了别人的代码摘出来的方法他的方法有三个,我就只看了我需要的
也可移步另外两个这是他的
第一次写这种东西有不对的地方请看到的多多指教