使用Python繪制詞云圖的詳細(xì)教程
更新時間：2025年01月02日 10:53:30 作者：一只蝸牛兒
詞云（Word Cloud）是一種數(shù)據(jù)可視化技術(shù),用于顯示文本數(shù)據(jù)中的頻繁單詞,在本教程中,我們將使用 Python 的 wordcloud 庫,結(jié)合 matplotlib 和 jieba 等工具,展示如何從文本數(shù)據(jù)生成詞云圖,需要的朋友可以參考下
引言
詞云（Word Cloud）是一種數(shù)據(jù)可視化技術(shù)，用于顯示文本數(shù)據(jù)中的頻繁單詞。它通過將出現(xiàn)頻率較高的詞匯以較大的字體顯示，頻率較低的詞匯則以較小的字體顯示，直觀地呈現(xiàn)出文本的關(guān)鍵詞。
在本教程中，我們將使用 Python 的 wordcloud 庫，結(jié)合 matplotlib 和 jieba 等工具，展示如何從文本數(shù)據(jù)生成詞云圖。
1. 安裝所需的庫
在開始之前，你需要確保已經(jīng)安裝了以下 Python 庫：
pip install wordcloud matplotlib jieba numpy pillow
wordcloud: 用于生成詞云圖。matplotlib: 用于顯示生成的詞云圖。jieba: 用于中文分詞。numpy: 用于處理數(shù)組數(shù)據(jù)。pillow: 用于圖像處理（可選，增強詞云圖的效果）。
2. 基本的詞云圖生成
我們從簡單的英文文本開始，展示如何生成詞云圖。
2.1 基本詞云生成
import matplotlib.pyplot as plt
from wordcloud import WordCloud
# 1. 準(zhǔn)備文本數(shù)據(jù)
text = "Python is a great programming language. Python is widely used for data science, web development, and automation."
# 2. 創(chuàng)建詞云對象
wordcloud = WordCloud(width=800, height=400, background_color="white").generate(text)
# 3. 顯示詞云圖
plt.figure(figsize=(10, 5))
plt.imshow(wordcloud, interpolation='bilinear')
plt.axis("off") # 不顯示坐標(biāo)軸
plt.show()
解釋：
WordCloud是用于生成詞云圖的類。我們通過設(shè)置width和height來調(diào)整圖像的大小，background_color="white"設(shè)置背景色為白色。.generate(text)方法用于從文本生成詞云。imshow()顯示生成的詞云圖，interpolation='bilinear'是為了讓圖像更平滑。
2.2 自定義詞云圖的外觀
你可以根據(jù)自己的需求定制詞云圖的外觀，例如設(shè)置不同的顏色、字體等。
示例：設(shè)置自定義顏色
wordcloud = WordCloud(width=800, height=400, background_color="black", colormap="coolwarm").generate(text)
plt.figure(figsize=(10, 5))
plt.imshow(wordcloud, interpolation='bilinear')
plt.axis("off")
plt.show()
colormap="coolwarm"：設(shè)置顏色映射，可以選擇不同的顏色方案，例如coolwarm，inferno，plasma等。
示例：設(shè)置自定義字體
wordcloud = WordCloud(font_path="/path/to/your/font.ttf", width=800, height=400, background_color="white").generate(text)
plt.figure(figsize=(10, 5))
plt.imshow(wordcloud, interpolation='bilinear')
plt.axis("off")
plt.show()
font_path可以設(shè)置一個自定義字體，特別適合中文詞云生成。
3. 中文詞云圖的生成
生成中文詞云圖的難點在于中文分詞。我們可以使用 jieba 庫對中文文本進(jìn)行分詞，然后再生成詞云圖。
3.1 中文分詞與詞云圖生成
import jieba
from wordcloud import WordCloud
import matplotlib.pyplot as plt
# 1. 準(zhǔn)備中文文本
text = "Python是一個非常強大的編程語言。Python廣泛應(yīng)用于數(shù)據(jù)科學(xué)、機器學(xué)習(xí)和人工智能等領(lǐng)域。"
# 2. 使用jieba進(jìn)行中文分詞
seg_list = jieba.cut(text)
word_list = " ".join(seg_list) # 將分詞結(jié)果轉(zhuǎn)換為字符串
# 3. 創(chuàng)建詞云對象并生成詞云
wordcloud = WordCloud(font_path="msyh.ttc", width=800, height=400, background_color="white").generate(word_list)
# 4. 顯示詞云圖
plt.figure(figsize=(10, 5))
plt.imshow(wordcloud, interpolation='bilinear')
plt.axis("off")
plt.show()
解釋：
jieba.cut(text)用于對中文文本進(jìn)行分詞。- 將分詞結(jié)果通過
" ".join(seg_list)拼接成一個字符串供詞云圖使用。 font_path="msyh.ttc"是指定一個支持中文的字體路徑（msyh.ttc是微軟雅黑字體，你可以根據(jù)需要選擇其他字體文件）。
4. 使用掩碼（Mask）生成定制形狀的詞云
掩碼（Mask）是指通過一張圖片來限定詞云圖的形狀，這樣生成的詞云就會以圖片的輪廓為形狀。
4.1 使用圖片掩碼
import numpy as np
from wordcloud import WordCloud
import matplotlib.pyplot as plt
from PIL import Image
# 1. 準(zhǔn)備文本數(shù)據(jù)
text = "Python is a versatile programming language. It is popular among data scientists, developers, and AI researchers."
# 2. 讀取掩碼圖像
mask_image = np.array(Image.open("cloud_shape.png"))
# 3. 創(chuàng)建詞云對象并生成詞云
wordcloud = WordCloud(width=800, height=400, background_color="white", mask=mask_image).generate(text)
# 4. 顯示詞云圖
plt.figure(figsize=(10, 5))
plt.imshow(wordcloud, interpolation='bilinear')
plt.axis("off")
plt.show()
解釋：
mask_image = np.array(Image.open("cloud_shape.png"))：我們加載一張圖像（例如心形、云形等），并將其轉(zhuǎn)換為 NumPy 數(shù)組作為掩碼。mask參數(shù)用于指定詞云圖的形狀。
注意：使用掩碼時，圖像的背景必須是白色或者透明，且圖像本身應(yīng)該是黑白色（黑色表示詞云區(qū)域，白色表示透明區(qū)域）。
5. 調(diào)整詞頻和生成自定義詞云
你可以通過手動調(diào)整單詞的頻率，來控制詞云圖中某些詞語的大小。
5.1 設(shè)置頻率字典
from wordcloud import WordCloud
import matplotlib.pyplot as plt
# 1. 創(chuàng)建詞頻字典
word_frequencies = {
"Python": 100,
"Java": 50,
"C++": 30,
"JavaScript": 70,
"Ruby": 10
}
# 2. 創(chuàng)建詞云對象
wordcloud = WordCloud(width=800, height=400, background_color="white").generate_from_frequencies(word_frequencies)
# 3. 顯示詞云圖
plt.figure(figsize=(10, 5))
plt.imshow(wordcloud, interpolation='bilinear')
plt.axis("off")
plt.show()
解釋：
- 使用
generate_from_frequencies()方法，我們可以根據(jù)頻率字典來生成詞云圖，其中字典的鍵是單詞，值是對應(yīng)的頻率。 - 詞云圖會根據(jù)頻率字典中提供的詞頻信息來顯示單詞，頻率較高的單詞會顯示得更大。
6. 保存詞云圖為文件
生成的詞云圖不僅可以在屏幕上顯示，還可以保存為圖像文件（如 PNG 或 JPG 格式）以供后續(xù)使用。
6.1 保存詞云圖
wordcloud = WordCloud(width=800, height=400, background_color="white").generate(text)
# 保存為文件
wordcloud.to_file("wordcloud_output.png")
解釋：
- 使用
to_file()方法將生成的詞云圖保存為文件。你可以指定保存的文件路徑和格式（如 PNG、JPG 等）。
7. 總結(jié)
在本文中，我們詳細(xì)介紹了如何使用 Python 和 wordcloud 庫生成詞云圖。我們學(xué)習(xí)了以下內(nèi)容：
- 基本詞云圖生成：通過簡單的文本生成詞云。
- 中文詞云圖生成：使用
jieba庫對中文文本進(jìn)行分詞，并生成詞云圖。 - 使用掩碼生成定制形狀的詞云：通過圖片掩碼來創(chuàng)建具有特定形狀的詞云。
- 自定義詞云：根據(jù)詞頻字典生成自定義的詞云圖。
- 保存詞云圖：將生成的詞云圖保存為圖像文件。
詞云是一種強大的可視化工具，可以幫助你從大量的文本數(shù)據(jù)中提取出最具代表性的詞匯。在實際應(yīng)用中，詞云廣泛用于分析社交媒體數(shù)據(jù)、評論數(shù)據(jù)等文本內(nèi)容。
到此這篇關(guān)于使用Python繪制詞云圖的詳細(xì)教程的文章就介紹到這了,更多相關(guān)Python繪制詞云圖內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
相關(guān)文章
pandas刪除某行或某列數(shù)據(jù)的實現(xiàn)示例
本文主要介紹了pandas刪除某行或某列數(shù)據(jù)的實現(xiàn)示例，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2022-08-08
這篇文章主要為大家詳細(xì)介紹了python openCV獲取人臉部分并存儲功能，具有一定的參考價值，感興趣的小伙伴們可以參考一下2019-08-08
這篇文章主要為大家詳細(xì)介紹了python pygame模塊編寫飛機大戰(zhàn)，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價值，感興趣的小伙伴們可以參考一下2018-11-11
pycharm遠(yuǎn)程連接服務(wù)器運行pytorch的過程詳解
這篇文章主要介紹了在Linux環(huán)境下使用Anaconda管理不同版本的Python環(huán)境,并通過PyCharm遠(yuǎn)程連接服務(wù)器來運行PyTorch的過程,包括安裝PyTorch、CUDA以及配置PyCharm遠(yuǎn)程開發(fā)環(huán)境的詳細(xì)步驟,需要的朋友可以參考下2025-02-02
在Python中os.fork()產(chǎn)生子進(jìn)程的例子
今天小編就為大家分享一篇在Python中os.fork()產(chǎn)生子進(jìn)程的例子，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-08-08
跟老齊學(xué)Python之關(guān)于循環(huán)的小伎倆
不管是while還是for，所發(fā)起的循環(huán)，在python編程中是經(jīng)常被用到的。特別是for，一般認(rèn)為，它要比while快，而且也容易寫（是否容易，可能因人而異，但是，執(zhí)行時間快，是的確的），因此在實踐中，for用的比較多點。2014-10-10
python調(diào)用百度語音識別實現(xiàn)大音頻文件語音識別功能
這篇文章主要為大家詳細(xì)介紹了python調(diào)用百度語音識別實現(xiàn)大音頻文件語音識別功能，具有一定的參考價值，感興趣的小伙伴們可以參考一下2018-08-08
Python實現(xiàn)隨機創(chuàng)建電話號碼的方法示例
這篇文章主要介紹了Python實現(xiàn)隨機創(chuàng)建電話號碼的方法,涉及Python隨機數(shù)運算相關(guān)操作技巧,需要的朋友可以參考下2018-12-12
Python結(jié)合SpeechRecognition和gTTS庫打造智能語音備忘錄
這篇文章主要為大家詳細(xì)介紹了如何利用Python的SpeechRecognition和gTTS庫,打造一款集語音輸入,自動轉(zhuǎn)文字,朗讀保存于一體的智能語音備忘錄工具,感興趣的小伙伴可以了解下2025-04-04
這篇文章主要介紹了Python pip配置國內(nèi)源的方法，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2020-02-02