Python制作词云的代码怎么写
代码艺术家
2024-04-13 14:03
短信预约 Python-IT技能 免费直播动态提醒
这篇文章将为大家详细讲解有关Python制作词云的代码怎么写,小编觉得挺实用的,因此分享给大家做个参考,希望大家阅读完这篇文章后可以有所收获。
导入必要的库
import numpy as np
import pandas as pd
from PIL import Image
from wordcloud import WordCloud, STOPWORDS
加载文本数据
text = "你的文本数据"
创建词云对象
# 设置词云参数
wordcloud = WordCloud(
width=800, # 宽度
height=600, # 高度
background_color="white", # 背景颜色
stopwords=STOPWORDS, # 停用词列表
max_words=200, # 最多显示的单词数
).generate(text)
生成词云
wordcloud.to_file("wordcloud.png") # 保存词云到文件
自定义词云形状
# 加载形状图像
mask = np.array(Image.open("shape.png"))
# 创建带形状的词云对象
wordcloud = WordCloud(
width=800,
height=600,
background_color="white",
mask=mask, # 形状掩码
max_words=200,
).generate(text)
# 生成词云
wordcloud.to_file("wordcloud_shaped.png")
使用字体和颜色
# 设置字体
font_path = "path/to/font.ttf"
wordcloud = WordCloud(
font_path=font_path,
).generate(text)
wordcloud.to_file("wordcloud_with_font.png")
# 设置颜色图
color_map = "path/to/color_map.png"
wordcloud = WordCloud(
colormap=color_map,
).generate(text)
wordcloud.to_file("wordcloud_with_color_map.png")
处理非英语文本
# 对非英语文本使用自定义停止词列表
stopwords = ["word1", "word2", "word3"]
wordcloud = WordCloud(
stopwords=stopwords,
).generate(text)
wordcloud.to_file("wordcloud_with_custom_stopwords.png")
# 使用正则表达式匹配非英语单词
import re
regex = r"[^ws]"
text = re.sub(regex, "", text)
wordcloud = WordCloud().generate(text)
wordcloud.to_file("wordcloud_non_english.png")
高级选项
- collocations: 允许单词对同时出现
- max_font_size: 设置最大字体大小
- scale: 缩放词云的比例
- contour_width: 设置轮廓宽度
- contour_color: 设置轮廓颜色
以上就是Python制作词云的代码怎么写的详细内容,更多请关注编程学习网其它相关文章!
免责声明:
① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。
② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341