基于Python編寫一個詞云制作程序
參數(shù)字典
自從做了熱榜的詞云之后,就越來越覺得詞云的表達力真的很強,所以合計是不是可以為WordCloud做一個界面,來更加直觀地操作。
既然以WordCloud為核心,那么界面的組件自然要和WordCloud的參數(shù)意義綁定,每個參數(shù)在界面中都對應一個提示組件,比如Lable,和一個輸入組件,比如Entry。此外,這些參數(shù)都是可設的,那么在對應一個組件的同時,也應該對應一個StringVar。
簡單起見,現(xiàn)在只挑選幾個必備的參數(shù),來演示這個邏輯的實現(xiàn)方法
| 參數(shù) | 說明 | 適用組件 | 參數(shù) |
|---|---|---|---|
| width | 詞云寬度 | Spinbox | 100-2000, 10 |
| height | 詞云高度 | Spinbox | 100-2000, 10 |
| scale | 縮放 | Spinbox | 0.5-10, 0.1 |
| 輸入文本 | 路徑對話框 | ||
| 輸出詞云 | 路徑對話框 | ||
| font_path | 字體文件路徑 | 路徑對話框 |
將這些參數(shù)封裝成一個字典,內(nèi)容如下
wcDct = {
"詞云寬度" : {
"Ctrl": ttk.Spinbox,
"paras" : {"from_":100, "to":2000, "increment":10},
"default":800,
"call" : "width"},
"詞云寬度" : {
"Ctrl": ttk.Spinbox,
"paras" : {"from_":100, "to":2000, "increment":10},
"default":450,
"call" : "height"},
"圖像縮放" : {
"Ctrl": ttk.Spinbox,
"paras" : {"from_":0.5, "to":10, "increment":0.1},
"default":1,
"call" : "scale"},
"輸入路徑" : {"Ctrl": DialogButton},
"輸出路徑" : {"Ctrl": DialogButton},
"字體路徑" : {"Ctrl": DialogButton, "call" : "font_path"},
}
其中,DialogButton是自定義的一個組件,主要包括一個按鈕和一個輸入框,點擊按鈕可以彈出文件對話框。
布局
有了這個字典,就可以據(jù)此來布局了
class DrawWords(ttk.Frame):
def __init__(self, master, **options):
super().__init__(master, **options)
self.pack()
self.words = None
self.initWidgets()
def initWidgets(self):
frm = ttk.Frame(self)
frm.pack(side=tk.LEFT, fill=tk.Y)
self.initPara(frm)
def setOneSpinBox(self, frm, key):
ttk.Label(frm, width=8, text=key).pack(side=tk.LEFT)
v = wcDct[key] # 組件參數(shù)
n = v["call"] # 調(diào)用名
self.spins[n] = v["ctrl"](frm, **v["paras"])
self.spins[n].set(v["default"])
self.spins[n].pack(side=tk.LEFT)
def setOneDiaButton(self, frmPara, key):
frm = ttk.Frame(frmPara)
frm.pack(side=tk.TOP, fill=tk.X)
ttk.Label(frm, width=8, text=key).pack(side=tk.LEFT)
v = wcDct[key]
n = v["call"] if 'call' in v else key
self.paths[n] = v["ctrl"](frm, **v['paras'])
self.paths[n].pack(side=tk.LEFT)
if 'default' in v:
self.paths[n].set(v['default'])
def initPara(self, frmPara):
self.spins = {}
frm = ttk.Frame(frmPara)
frm.pack(side=tk.TOP, fill=tk.X)
for key in ["詞云寬度", "詞云高度"]:
self.setOneSpinBox(frm, key)
self.paths = {}
for key in ["輸入路徑", "輸出路徑", "字體路徑"]:
self.setOneDiaButton(frmPara, key)
frm = ttk.Frame(frmPara)
frm.pack(side=tk.TOP, fill=tk.X)
ttk.Button(frm, text="輸出詞云",
command=self.genWordCloud).pack(side=tk.LEFT)
在布局過程中,對SpinBox類型的變量和DialogButton型的變量分別做了不同的處理,并且為它們分別設計了全局變量字典self.spins和self.paths,以便于調(diào)用。
布局結(jié)果如下

其中,genWordCloud用于詞云輸出,考慮到分詞的準確性,這里仍然采用jieba進行分詞,并為分詞專門設計一個函數(shù),其最終實現(xiàn)如下。
def splitWords(self):
p = self.paths["輸入路徑"].get()
with open(p, encoding='utf8') as f:
text = f.read()
words = jieba.lcut(text)
self.words = [w for w in words if len(w)>1] # 取出長度大于1的詞
def genWordCloud(self):
dct = {}
for key in ['width', 'height', 'font_path']:
if key in self.spins:
dct[key] = int(self.spins[key].get())
if key in self.paths:
dct[key] = self.paths[key].get()
cloud = WordCloud(**dct)
self.splitWords()
cloud.generate(" ".join(self.words))
p = self.paths["輸出路徑"].get()
if not (p.endswith('.png') or p.endswith('.svg')):
p = p+".png"
cloud.to_file(p)
測試結(jié)果
最后,按照慣例,做一下測試
if __name__ == "__main__":
root = tk.Tk()
DrawWords(root).pack(side=tk.TOP, fill=tk.BOTH)
root.mainloop()
效果如下

到此這篇關于基于Python編寫一個詞云制作程序的文章就介紹到這了,更多相關python詞云內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家!
相關文章
Python編程二分法實現(xiàn)冒泡算法+快速排序代碼示例
這篇文章主要介紹了Python編程二分法實現(xiàn)冒泡算法+快速排序代碼示例,具有一定借鑒價值,需要的朋友可以參考下2018-01-01
解決編碼問題:UnicodeDecodeError: 'utf-8' codec
這篇文章主要介紹了快速解決編碼問題:UnicodeDecodeError: 'utf-8' codec can't decod,具有很好的參考價值,希望對大家有所幫助。如有錯誤或未考慮完全的地方,望不吝賜教2022-05-05
pandas 使用apply同時處理兩列數(shù)據(jù)的方法
下面小編就為大家分享一篇pandas 使用apply同時處理兩列數(shù)據(jù)的方法,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2018-04-04

