问题遇到的现象和发生背景
import pandas as pd
import numpy as np
import jieba
from wordcloud import WordCloud, ImageColorGenerator
import matplotlib.pyplot as plt
from PIL import Image
读取数据
data = pd.read_csv(open("D:/luolan/2018-2022年一二产业增加值.csv"))
提取地区列数据
regions = data['地区']
将地区名称拼接为一个字符串
text = ' '.join(regions)
使用jieba进行分词
seg_list = jieba.lcut(text)
将分词结果转换为空格分隔的字符串
seg_text = ' '.join(seg_list)
读取背景图片
background_image = "D:/luolan/background1.png"
mask = np.array(Image.open(background_image))
生成词云图
wordcloud = WordCloud(mask=mask, font_path="C:/Windows/Fonts/simsunb.ttf", background_color="white").generate(seg_text)
根据背景图片设置颜色
image_colors = ImageColorGenerator(mask)
plt.imshow(wordcloud.recolor(color_func=image_colors), interpolation='bilinear')
plt.axis("off")
plt.show()
操作环境、软件版本等信息
尝试过的解决方法
报错如下
选择删除解决
我想要达到的结果
将地区生成词云图