Python是一種很受歡迎的編程語言,不僅在科學計算領域得到了廣泛應用,也在文本分析和可視化領域發揮了重要作用。一種特別受歡迎的Python可視化工具是詞云圖。在這篇文章中,我們將討論如何使用Python創建詞云圖。
import wordcloud from wordcloud import WordCloud, STOPWORDS import matplotlib.pyplot as plt import numpy as np from PIL import Image # 讀取文本文件內容 text = open('text.txt', 'r').read() # 設置停止詞 stopwords = set(STOPWORDS) stopwords.add('said') # 讀取矩形圖像文件 path = 'image.png' mask = np.array(Image.open(path)) # 配置WordCloud參數 wc = WordCloud(background_color="white", max_words=2000, mask=mask, stopwords=stopwords, contour_width=3, contour_color='steelblue') # 生成詞云圖 wc.generate(text) # 保存詞云圖 wc.to_file("result.png") # 顯示詞云圖 plt.figure(figsize=[10,8]) plt.imshow(wc, interpolation='bilinear') plt.axis("off") plt.show()
在這個代碼塊里,我們首先導入wordcloud庫,然后通過open()函數打開文本文件。我們還使用了STOPWORDS集合,并添加了一個停用詞“said”。接下來,我們讀取了矩形圖像文件,使用np.array()函數轉換為NumPy數組。隨后,我們創建了一個WordCloud類的實例,并根據需求進行了相應的配置。最后,我們生成了詞云圖,并將其保存為result.png文件。接著使用matplotlib.pyplot庫顯示生成的詞云圖。最后,我們通過plt.imshow()和plt.axis()函數設置不顯示坐標軸,最后通過plt.show()函數顯示帶有詞云的圖像。
總結起來,使用Python創建詞云圖并不難,只需按照上面的步驟進行操作即可。你也可以使用不同的文件以及圖像進行測試,直到你得到你滿意的詞云圖。希望這篇文章能夠為Python初學者提供幫助,也讓更多的人了解Python的強大特性。