|
|
from tkinter import Image
|
|
|
import numpy as np
|
|
|
from wordcloud import WordCloud
|
|
|
import matplotlib.pyplot as plt
|
|
|
from imageio import imread
|
|
|
from PIL import Image
|
|
|
import random
|
|
|
import os
|
|
|
|
|
|
def get_random_color(*args, **kwargs):
|
|
|
colors = ["rgb(255, 182, 193)", # 淡粉色
|
|
|
"rgb(173, 216, 230)", # 淡蓝色
|
|
|
"rgb(252, 210, 110)"] # 淡黄色
|
|
|
return random.choice(colors)
|
|
|
|
|
|
def pink_color_func(*args, **kwargs):
|
|
|
return "rgb(255, 182, 193)" # 淡粉色
|
|
|
|
|
|
def orange_color_func(*args, **kwargs):
|
|
|
return "rgb(252, 210, 110)" # 淡黄色
|
|
|
|
|
|
# 生成词云图，可以通过添加style参数选择不同的风格
|
|
|
def generate_wordcloud(text, font_path, width, height, style='basic'):
|
|
|
current_dir = os.path.dirname(os.path.abspath(__file__))
|
|
|
if style == 'basic':
|
|
|
image_path = os.path.join(current_dir, 'image', 'love.png')
|
|
|
mask = np.array(Image.open(image_path))
|
|
|
wordcloud = WordCloud(mask=mask, font_path=font_path, width=width, background_color='white', height=height).generate(text)
|
|
|
elif style == 'jiyi':
|
|
|
image_path = os.path.join(current_dir, 'image', 'ji1.png')
|
|
|
mask = np.array(Image.open(image_path))
|
|
|
wordcloud = WordCloud(mask=mask, color_func=pink_color_func, font_path=font_path, width=width, background_color='white', height=height).generate(text)
|
|
|
elif style == '537':
|
|
|
image_path = os.path.join(current_dir, 'image', '537.png')
|
|
|
mask = np.array(Image.open(image_path))
|
|
|
wordcloud = WordCloud(mask=mask, color_func=orange_color_func, font_path=font_path, width=width, background_color='white', height=height).generate(text)
|
|
|
elif style == 'family':
|
|
|
image_path = os.path.join(current_dir, 'image', 'family.png')
|
|
|
mask = np.array(Image.open(image_path))
|
|
|
wordcloud = WordCloud(mask=mask, color_func=get_random_color, font_path=font_path, width=width, height=height, background_color='white').generate(text)
|
|
|
else:
|
|
|
wordcloud = WordCloud(font_path=font_path, width=width, height=height, mode='RGBA', relative_scaling=0.5).generate(text)
|
|
|
|
|
|
# 显示词云
|
|
|
plt.figure(figsize=(10, 5))
|
|
|
plt.imshow(wordcloud, interpolation='bilinear')
|
|
|
plt.axis('off')
|
|
|
plt.show()
|
|
|
|
|
|
|
