幾行代碼教你使用python製作詞雲（四）

幾行代碼教你使用python製作詞雲（四）

這裏給大家介紹wordcloud中勾勒輪廓參數，contour_width和contour_color設置輪廓寬度和顏色（文章最後附有案例完整代碼）

1.準備圖片和數據集

案例數據爲蠟筆小新的介紹

圖片也爲蠟筆小新

2.代碼分析

導入詞雲製作庫wordcloud

# 導入詞雲製作庫wordcloud
import wordcloud

將外部文件包含的文本保存在string變量中

# 將外部文件包含的文本保存在string變量中
string = open('./data/XiaoXin.txt',encoding='utf-8').read()

導入imageio庫中的imread函數，並用這個函數讀取本地圖片，作爲詞雲形狀圖片

# 導入imageio庫中的imread函數，並用這個函數讀取本地圖片，作爲詞雲形狀圖片
import imageio
mk = imageio.imread("./input/xiaoxin.jpg")

構建詞雲對象

# 構建詞雲對象w，注意增加參數contour_width和contour_color設置輪廓寬度和顏色
w = wordcloud.WordCloud(background_color="white",
                        mask=mk,
                        contour_width=1,
                        contour_color='steelblue')

將string變量傳入w的generate()方法，給詞雲輸入文字

 # 將string變量傳入w的generate()方法，給詞雲輸入文字
w.generate(string)

在控制檯繪製並輸出圖像（這一步可要可不要）

# 在控制檯繪製並輸出圖像
import matplotlib.pyplot as plt  #繪圖
plt.imshow(w,interpolation="bilinear")
plt.axis("off")

將詞雲圖片導出到指定文件夾

# 將詞雲圖片導出到當前文件夾
w.to_file('./output/xiaoxin.png')

效果圖

3.完整代碼

# -*- coding: utf-8 -*-
"""
Created on Mon Jun  2 10:24:07 2020

@author: Jay-L
"""

# 導入詞雲製作庫wordcloud
import wordcloud

# 將外部文件包含的文本保存在string變量中
string = open('./data/XiaoXin.txt',encoding='utf-8').read()

# 導入imageio庫中的imread函數，並用這個函數讀取本地圖片，作爲詞雲形狀圖片
import imageio
mk = imageio.imread("./input/xiaoxin.jpg")

# 構建詞雲對象w，注意增加參數contour_width和contour_color設置輪廓寬度和顏色
w = wordcloud.WordCloud(background_color="white",
                        mask=mk,
                        contour_width=1,
                        contour_color='steelblue')

# # 將string變量傳入w的generate()方法，給詞雲輸入文字
w.generate(string)

# 在控制檯繪製並輸出圖像
import matplotlib.pyplot as plt  #繪圖
plt.imshow(w,interpolation="bilinear")
plt.axis("off")

# 將詞雲圖片導出到當前文件夾
w.to_file('./output/xiaoxin.png')