python词云设计之--爱心型

效果图：

代码实现：

import numpy as np
import jieba
from PIL import Image
from wordcloud import WordCloud, STOPWORDS
import matplotlib.pyplot as plt

def draw_word_cloud(word):
words = jieba.cut(word)
wordstr = " ".join(words)
sw = set(STOPWORDS)
sw.add("ok")
mask = np.array(Image.open('2.jpg'))
wc = WordCloud(
　　font_path='C:/Windows/Fonts/simhei.ttf', # 设置字体格式
　　background_color="white",
　　mask=mask,
　　max_words=200,
　　max_font_size=100,
　　stopwords=sw,
　　scale=4,
　　).generate(wordstr)

# 显示词云图
plt.imshow(wc)
plt.axis("on")
plt.show()
# 保存词云图
wc.to_file('result.jpg')

if __name__ == "__main__":
with open("test2.txt", "rb") as f:
word = f.read()
draw_word_cloud(word)

附件：背景图2.jpg