Python数据可视化：WordCloud+GUI
2019-07-26 本文已影响37人
洋阳酱
之前写过一篇WordCloud入门，里面写了生成WordCloud的一般方法，包括参数设计和形状修改等等
标签云效果图--基础
在上一次的基础上，这次用easygui做了一个界面
先展示一下界面，代码在最后
输入文字信息
选择需要输入的内容
显示所选文字
生成WordCloud
生成WordCloud
这是默认图像，可选择是否要继续优化
继续优化参数：进入后续优化
直接保存图像：退出程序，图像保存在代码相对应的文件夹
继续优化
设置指定形状
选择你要的图片
是否和图片颜色保持一致
参数设置
根据上述选择结果，这里可以看到你选择的轮廓参照文件和图片顺色情况，你依旧可以改动这些参数
根据刚刚的选择，我们得到新的WordCloud
此时的图像显示会模糊一些，实际保存的图像太大了，在显示的时候，我降低了像素（不影响实际结果）
在这个结果上，可以继续点击继续优化参数，重复以上步骤，设置新的参数
调整完毕后，点击直接保存图像
此时，就会保存高清的图像到你的文件夹
实际图像
代码说明
这个程序，还是有很多可以优化的地方
- 剔除不要的单词
- 预设长方形、正方形、圆形
- 将颜色设为选项，而不是需要填写
- 后续慢慢优化
代码主要思路
用fileopenbox打开txt的对话框
import easygui as g
path_text = g.fileopenbox(title = '请选择需要输入的内容（后缀为.txt）', default="*.txt")
显示txt内容
msg='文件内容如下：'
title='显示文件内容'
text = open(path_text, encoding='utf-8').read()
g.textbox(msg,title,text)
精确切割中文字符（这样保证中英文都可以使用）
如果只要用英文，那这一步可以省略
text = ' '.join(jieba.cut(text, cut_all = False))
设置默认WordCloud参数
fieldValues = [
'C:/Windows/Fonts/simfang.ttf', # 字体路径
100, # 显示词数
100, # 最大字号
'white', # 背景色
0, # 轮廓宽度
'steelblue', # 轮廓颜色
'False', # 轮廓参照
'False' # 和图片顺色
]
wc = SetWordCloud(fieldValues)
生成初步图像
wc.generate(text)
wc.to_file('image.png') # 储存图像
判断是否要优化
def IfContinue(WC_continue, path_image, fieldValues) # 是否要继续优化
不优化，直接保存高清图像
优化，继续下一步
如果指定了背景形状
def Para_mask(fieldValues) # 指定图片形状和颜色
def Para(values) # 修改其他参数
def SetWordCloudMask(mask_coloring, fieldValues)
如果不指定背景形状
def Para(values) # 修改其他参数
def SetWordCloud(fieldValues)
根据参数，生成图像
完整代码
import os
from os import path
# from matplotlib import pyplot as plt
import jieba
import easygui as g
import numpy as np
from PIL import Image
from wordcloud import WordCloud, STOPWORDS, ImageColorGenerator
# 用户设定初始值
def Para(values):
msg = "你可以修改下面的默认参数"
title = "参数设置"
fieldNames = [
"字体路径" , # fieldValues[0]
"显示词数", # fieldValues[1]
"最大字号" , # fieldValues[2]
"背景色", # fieldValues[3]
"轮廓宽度", # fieldValues[4]
"轮廓颜色", # fieldValues[5]
"轮廓参照", # fieldValues[6]
"和图片顺色" # fieldValues[7]
]
fieldValues = []
fieldValues = g.multenterbox (msg, title, fieldNames,values = values)
return fieldValues
# 用户设定初始值-背景轮廓和颜色
def Para_mask(fieldValues):
d = path.dirname(__file__) if "__file__" in locals() else os.getcwd()
mask_image = path.join(d,'mask.jpg')
if_mask = g.buttonbox("需要指定形状吗？\n默认是长方形\n你可以输入一张图片作为轮廓", image = mask_image,choices=("需要","不需要"))
if if_mask == "需要":
fieldValues[6] = g.fileopenbox(msg = '后缀为图片格式', title = '设置mask路径', default="*.png")
#mask = np.array(Image.open(path_mask))
color_image = path.join(d,'colored.png')
if_color = g.buttonbox("需要和图片顺色吗？左边是不顺色，中间是顺色，右边是原图", image = color_image, choices=("需要","不需要"))
if if_color == "需要":
fieldValues[7] = 'True'
else:
fieldValues[7] = 'False'
else:
fieldValues[6] = 'False'
return fieldValues
# 是否需要继续优化参数
def IfContinue(WC_continue, path_image, fieldValues):
if WC_continue == "继续优化参数" :
fieldValues = Para_mask(fieldValues)
fieldValues = Para(fieldValues)
if fieldValues[6] == 'False':
wc = SetWordCloud(fieldValues)
wc.generate(text)
wc.to_file('image.png') # 储存图像
WC_continue = g.buttonbox(title="显示图像",msg='是否需要优化', image = path_image, choices=("继续优化参数","直接保存图像"))
IfContinue(WC_continue, path_image, fieldValues)
else:
# 读取图片文件
mask_coloring = np.array(Image.open(fieldValues[6]))
wc = SetWordCloudMask(mask_coloring, fieldValues)
wc.generate(text)
if fieldValues[7] == 'False':
wc.to_file('image.png') # 储存图像
# 调整图像显示大小
im = Image.open(path_image)
(x,y) = im.size #read image size
y_s = 500 #define standard width
x_s = int(x*y_s/y) #calc height based on standard width
im = im.resize((x_s,y_s)) #resize image with high-quality
im .save('image.png')
WC_continue = g.buttonbox(title="显示图像",msg='是否需要优化', image = path_image, choices=("继续优化参数","直接保存图像"))
wc.to_file('image.png') # 储存图像
IfContinue(WC_continue, path_image, fieldValues)
else:
# 根据图片，创建颜色
image_colors = ImageColorGenerator(mask_coloring)
# 储存图像
wc.recolor(color_func = image_colors).to_file('image.png')
# 调整图像显示大小
im = Image.open(path_image)
(x,y) = im.size #read image size
y_s = 500 #define standard width
x_s = int(x*y_s/y) #calc height based on standard width
im = im.resize((x_s,y_s)) #resize image with high-quality
im .save('image.png')
WC_continue = g.buttonbox(title = "显示图像",msg = '是否需要优化', image = path_image, choices=("继续优化参数","直接保存图像"))
wc.recolor(color_func = image_colors).to_file('image.png')
IfContinue(WC_continue, path_image, fieldValues)
# 设置WordCloud参数，生成词云
def SetWordCloud(fieldValues):
wc = WordCloud(
font_path = fieldValues[0], #字体路径
scale = 2,
max_words = int(fieldValues[1]), #最多词个数
max_font_size = int(fieldValues[2]), #最大字号
background_color = fieldValues[3], #背景色
contour_width = int(fieldValues[4]), #设置轮廓宽度
contour_color = fieldValues[5] #设置轮廓颜色
)
return wc
# 设置WordCloud参数，生成词云
def SetWordCloudMask(mask_coloring, fieldValues):
wc = WordCloud(
font_path = fieldValues[0], #字体路径
scale = 2,
max_words = int(fieldValues[1]), #最多词个数
max_font_size = int(fieldValues[2]), #最大字号
background_color = fieldValues[3], #背景色
contour_width = int(fieldValues[4]), #设置轮廓宽度
contour_color = fieldValues[5], #设置轮廓颜色
mask = mask_coloring
)
return wc
# 选择要输入的文件
path_text = g.fileopenbox(title = '请选择需要输入的内容（后缀为.txt）', default="*.txt")
# 显示文件
msg='文件内容如下：'
title='显示文件内容'
text = open(path_text, encoding='utf-8').read()
g.textbox(msg,title,text)
# 精确切割中文字符
text = ' '.join(jieba.cut(text, cut_all = False))
# 默认参数
fieldValues = [
'C:/Windows/Fonts/simfang.ttf',
100,
100,
'white',
0,
'steelblue',
'False',
'False'
]
wc = SetWordCloud(fieldValues)
wc.generate(text)
wc.to_file('image.png') # 储存图像
# 获取当前文件路径
d = path.dirname(__file__) if "__file__" in locals() else os.getcwd()
# 获取文本txt的路径（txt和代码在一个路径下面）
path_image = path.join(d,'image.png')
WC_continue = g.buttonbox(title="显示图像",msg='是否需要优化', image = path_image, choices=("继续优化参数","直接保存图像"))
IfContinue(WC_continue, path_image, fieldValues)
···
