python 词云 调整

笔记2024-01-1315 人已阅来源:网络

Python词云库是一种有趣且有用的数据可视化工具,它可以生成各种文本形式的数据集词云图。然而,如何调整生成的词云图,使其更好看,更符合个人喜好呢?这里给大家分享一些简单的方法和技巧。

from wordcloud import WordCloud
# 先生成一个基本的词云图
wc = WordCloud()
wc.generate("hello world")
# 调整词云图的尺寸和背景颜色
wc.width = 800
wc.height = 400
wc.background_color = "white"
# 调整词云图中的单词颜色和形状
import numpy as np
from PIL import Image
mask = np.array(Image.open("mask.jpg"))
wc = WordCloud(mask=mask, color_func=lambda *args, **kwargs: "black")
wc.generate("hello world")
# 调整词云图中的单词位置和字体大小
import random
def random_position(word, font_size, position, orientation, **kwargs):
# 随机改变每个单词的位置
return random.randint(-10, 10), random.randint(-10, 10)
wc = WordCloud(font_path="fonts/simhei.ttf", width=800, height=400, 
background_color="white", color_func=lambda *args, **kwargs: "black",
prefer_horizontal=0.9, min_font_size=4, max_font_size=100)
wc.generate("hello world")
# 最后生成并显示词云图
import matplotlib.pyplot as plt
plt.imshow(wc, interpolation="bilinear")
plt.axis("off")
plt.show()

通过以上几个例子,大家可以了解到如何通过改变词云图尺寸、颜色、形状、单词位置及字体大小等参数调整词云图样式。当然,这只是冰山一角,Python词云库还有很多高级特性等待大家去探索。