Python實(shí)現(xiàn)繪制自定義形狀的詞云示例

更新時(shí)間：2024年04月30日 11:42:56 作者：早八起得來

這篇文章主要介紹了Python實(shí)現(xiàn)繪制自定義形狀的詞云示例，通過將一段文本中出現(xiàn)頻率高的單詞按其出現(xiàn)頻率大小以及顏色深淺排列成一個(gè)詞云圖形，從而更好地展示文本中的信息，需要的朋友可以參考下

前言

本文為分總結(jié)構(gòu)，有特定需求的可以查閱前部分分結(jié)構(gòu)的對(duì)應(yīng)板塊，最后的總結(jié)不懂的可以在分板塊查閱解釋。分板塊分別有引用的庫(kù)、閱讀文本、分詞并設(shè)置停用詞、設(shè)置png掩膜、字體設(shè)置、生成詞云圖，感謝您點(diǎn)開這篇分享，祝順利。

一、引用的庫(kù)

from wordcloud import WordCloud, STOPWORDS
import matplotlib.pyplot as plt
import numpy as np
import jieba.posseg as pseg
from collections import Counter
import PIL.Image as Image
from matplotlib import colors

確保已安裝以上的庫(kù)，不然運(yùn)行會(huì)報(bào)錯(cuò)

#安裝庫(kù)可以用清華的鏡像網(wǎng)站（可能會(huì)更新，可以上官網(wǎng)查詢地址）

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple some-package

二、閱讀文本(讓python閱讀)

# 閱讀文本（這里yourfile.txt，根據(jù)文本所在具體位置進(jìn)行設(shè)置）
text = open("yourfile.txt", encoding="utf-8").read()  
words = pseg.cut(text)

這里"yourfile.txt"，根據(jù)文本所在具體位置進(jìn)行設(shè)置

三、分詞并設(shè)置停用詞

# 按指定長(zhǎng)度和詞性提取詞
report_words = []
for word, flag in words:
    if (len(word) >= 2) and ('n' in flag): #這里設(shè)置統(tǒng)計(jì)的字?jǐn)?shù)
        report_words.append(word)
# 設(shè)置停用詞
stopwords = set(STOPWORDS)
stopwords.update(["的", "感謝", "我代表", "以上", "報(bào)告", "表示誠(chéng)摯感謝","戰(zhàn)略"])
# 去除停用詞
report_words = [word for word in report_words if word not in stopwords]
# 統(tǒng)計(jì)高頻詞匯
result = Counter(report_words).most_common(200) #詞的個(gè)數(shù)
# 建立詞匯字典
content = dict(result)
#輸出詞頻統(tǒng)計(jì)結(jié)果
for i in range(50):
    word,flag=result[i]
    print("{0:<10}{1:>5}".format(word,flag))

len（word）設(shè)置的是詞長(zhǎng)，想要提取兩個(gè)兩個(gè)字的就設(shè)置2，三個(gè)三個(gè)字的就設(shè)置3（以此類推）

result = Counter(report_words).most_common(200) 這里的200是指統(tǒng)計(jì)200個(gè)詞用于繪制，可以根據(jù)需求設(shè)置

四、設(shè)置png掩膜

#設(shè)置png掩膜（yourfile.png根據(jù)實(shí)際路徑進(jìn)行替換）
background = Image.open("yourfile.png")
mask = np.array(background)

yourfile.png根據(jù)實(shí)際路徑進(jìn)行替換

如果輸出結(jié)構(gòu)還是長(zhǎng)方形（正方形），應(yīng)該是png圖片“有問題”，可以嘗試以下的處理

1.用p圖軟件把圖片改成純黑色（可能別的也可以，我沒試過）

2.用以下代碼把白色背景改為透明

# 如果當(dāng)前位深是32的話，可以不用寫轉(zhuǎn)RGBA模式的這一句，但是寫上也沒啥問題
# 從RGB（24位）模式轉(zhuǎn)成RGBA（32位）模式
img = Image.open("yourfile.png").convert('RGBA')
W, L = img.size
white_pixel = (0, 0, 0, 0)  # 白色
for h in range(W):
    for i in range(L):
        if img.getpixel((h, i)) == white_pixel:
            img.putpixel((h, i), (255, 255, 255, 0))  # 設(shè)置透明
img.save("yourfile_new.png")  # 自己設(shè)置保存地址

這里有兩個(gè)參數(shù)需要修改

yourfile.png根據(jù)實(shí)際路徑進(jìn)行替換，yourfile_new.png（這是修改后圖片）根據(jù)實(shí)際路徑進(jìn)行替換

五、字體設(shè)置

# 設(shè)置字體樣式路徑
font_path = r"C:\Windows\Fonts\STLITI.TTF"
# 設(shè)置字體大小
max_font_size =200
min_font_size =10
# 建立顏色數(shù)組，可更改顏色
color_list = ['#FF274B']
# 調(diào)用顏色數(shù)組
colormap = colors.ListedColormap(color_list)

字體樣式：一般都在這個(gè)路徑，可以自己根據(jù)需求修改或者下載想要的字體

字體大?。鹤畲蠛妥钚「鶕?jù)需求更改

字體顏色：可以不要這行代碼（默認(rèn)設(shè)置），也可以根據(jù)需求設(shè)置一種或多種顏色（我這里只設(shè)置了一種）

六、生成詞云圖

# 生成詞云
wordcloud = WordCloud(scale=4,                         #輸出清晰度
                      font_path=font_path,             #輸出路徑
                      colormap=colormap,               #字體顏色
                      width=1600,                      #輸出圖片寬度
                      height=900,                      #輸出圖片高度
                      background_color='white',        #圖片背景顏色
                      stopwords=stopwords,             #停用詞
                      mask=mask,                       #掩膜
                      max_font_size=max_font_size,     #最大字體大小
                      min_font_size=min_font_size)     #最小字體大小
wordcloud.generate_from_frequencies(content)
# 使用 matplotlib 顯示詞云
plt.imshow(wordcloud, interpolation='bilinear')
plt.axis('off')
plt.show()
# 保存詞云圖
wordcloud.to_file("wordcloud.png")

如果前面參數(shù)都是按著我的來設(shè)置的話，這里直接復(fù)制粘貼就好

總結(jié)

from wordcloud import WordCloud, STOPWORDS
import matplotlib.pyplot as plt
import numpy as np
import jieba.posseg as pseg
from collections import Counter
import PIL.Image as Image
from matplotlib import colors
# 閱讀文本（這里yourfile.txt，根據(jù)文本所在具體位置進(jìn)行設(shè)置）
text = open("yourfile.txt", encoding="utf-8").read()
words = pseg.cut(text)
# 按指定長(zhǎng)度和詞性提取詞
report_words = []
for word, flag in words:
    if (len(word) >= 2) and ('n' in flag): #這里設(shè)置統(tǒng)計(jì)的字?jǐn)?shù)
        report_words.append(word)
# 統(tǒng)計(jì)高頻詞匯
result = Counter(report_words).most_common(200) #詞的個(gè)數(shù)
# 建立詞匯字典
content = dict(result)
#輸出詞頻統(tǒng)計(jì)結(jié)果
for i in range(50):
    word,flag=result[i]
    print("{0:<10}{1:>5}".format(word,flag))
# 設(shè)置停用詞
stopwords = set(STOPWORDS)
stopwords.update(["的", "感謝", "我代表", "以上", "報(bào)告", "表示誠(chéng)摯感謝","戰(zhàn)略"])
#設(shè)置png掩膜（yourfile.png根據(jù)實(shí)際路徑進(jìn)行替換）
background = Image.open("yourfile.png").convert('RGB')
mask = np.array(background)
'''
# 如果當(dāng)前位深是32的話，可以不用寫轉(zhuǎn)RGBA模式的這一句，但是寫上也沒啥問題
# 從RGB（24位）模式轉(zhuǎn)成RGBA（32位）模式
img = Image.open("yourfile.png").convert('RGBA')
W, L = img.size
white_pixel = (0, 0, 0, 0)  # 白色
for h in range(W):
    for i in range(L):
        if img.getpixel((h, i)) == white_pixel:
            img.putpixel((h, i), (255, 255, 255, 0))  # 設(shè)置透明
img.save("yourfile_new.png")  # 自己設(shè)置保存地址
'''
# 設(shè)置字體樣式路徑
font_path = r"C:\Windows\Fonts\STLITI.TTF"
# 設(shè)置字體大小
max_font_size =200
min_font_size =10
# 建立顏色數(shù)組，可更改顏色
color_list = ['#FF274B']
# 調(diào)用顏色數(shù)組
colormap = colors.ListedColormap(color_list)
# 生成詞云
wordcloud = WordCloud(scale=4,                         #輸出清晰度
                      font_path=font_path,             #輸出路徑
                      colormap=colormap,               #字體顏色
                      width=1600,                      #輸出圖片寬度
                      height=900,                      #輸出圖片高度
                      background_color='white',        #圖片背景顏色
                      stopwords=stopwords,             #停用詞
                      mask=mask,                       #掩膜
                      max_font_size=max_font_size,     #最大字體大小
                      min_font_size=min_font_size)     #最小字體大小
wordcloud.generate_from_frequencies(content)
# 使用 matplotlib 顯示詞云
plt.imshow(wordcloud, interpolation='bilinear')
plt.axis('off')
plt.show()
# 保存詞云圖
wordcloud.to_file("wordcloud.png")