赘婿词云图制作

from matplotlib import pyplot as plt
from wordcloud import WordCloud,ImageColorGenerator
import jieba

fp = open('赘婿.txt','r')
text = fp.read()
cut_word = " ".join(jieba.cut(text))

stopwords = set()
content = [line.strip() for line in open('hit_stopwords.txt','r').readlines()]
stopwords.update(content)

background_image = plt.imread('赘婿.jpg')

wc = WordCloud(background_color = 'white',
               font_path = r'/System/Library/Fonts/Supplemental/Songti.ttc',
               stopwords = stopwords,
               mask = background_image,
               max_words=2000)
wc.generate(cut_word)

#改变字体颜色
img_colors = ImageColorGenerator(background_image)
#字体颜色为背景图片的颜色
wc.recolor(color_func=img_colors)

plt.imshow(wc, interpolation='bilinear')
plt.axis('off') #取消坐标轴
plt.show()