词云图python代码

生成词云图的python代码如下：

import numpy as np
import pandas as pd
from PIL import Image
from wordcloud import WordCloud, STOPWORDS, ImageColorGenerator
import matplotlib.pyplot as plt

# 读取文本文件
text = open('text.txt', 'r', encoding='utf-8').read()

# 加载背景图片
image = np.array(Image.open('background.png'))

# 设置停用词
stopwords = set(STOPWORDS)
stopwords.add('said')

# 生成词云图
wordcloud = WordCloud(background_color='white', max_words=2000, mask=image, stopwords=stopwords, contour_width=3, contour_color='steelblue')
wordcloud.generate(text)

# 从图片中提取颜色
image_colors = ImageColorGenerator(image)

# 显示词云图
plt.figure(figsize=[10, 10])
plt.imshow(wordcloud.recolor(color_func=image_colors), interpolation='bilinear')
plt.axis('off')
plt.show()

# 保存词云图
wordcloud.to_file('wordcloud.png')

需要注意的是，要生成词云图需要先安装`wordcloud`库和`Pillow`库。