用Python讀紅樓夢之——二、詞雲美化

原文鏈接

咱們書接上回，講到我們使用Python把紅樓夢中的核心辭彙給繪畫出來了，但是，紅樓夢這麼唯美的書，給我們烏漆麻黑的搞了一張詞雲，寶寶們肯定接受不了。

我們先來找照片，這張照片不錯，不過還是要處理一下，把背景處理稱為白色，這個時候，PS工具可以簡單地達到這個效果。

接著，我們就要來寫程序了，老規矩，我們先來導入包：

from scipy.misc import imread

import matplotlib.pyplot as plt

from wordcloud import WordCloud, ImageColorGenerator

把圖片讀取進去，然後重新設置一下WordCloud對象，設置mask為我們的圖片。

bimg = imread("D:\賈寶玉.png")

wordcloud = WordCloud(

background_color="white",

mask=bimg, font_path=D:\simhei.ttf

)

接著，我們就可以美美地繪製詞雲了。

wordcloud = wordcloud.fit_words(segStat.head(1000).itertuples(index=False))

bimgColors = ImageColorGenerator(bimg)

plt.axis("off")

plt.imshow(wordcloud.recolor(color_func=bimgColors))

plt.show()

長腿歐巴雖然好，但是畫起來，很明顯我們的辭彙分成兩大部分了，這樣子會分散我們的注意力，因此，我們使用人物作為背景繪製詞雲的時候，最好只是使用上半身，使用頭像就更好了。因此，長腿歐巴拜拜。

bimg = imread("D:\賈寶玉2.png")

wordcloud = WordCloud(

background_color="white",

mask=bimg, font_path=D:\simhei.ttf

)

wordcloud = wordcloud.fit_words(segStat.head(1000).itertuples(index=False))

bimgColors = ImageColorGenerator(bimg)

plt.axis("off")

plt.imshow(wordcloud.recolor(color_func=bimgColors))

plt.show()

這樣子畫出來的詞雲，是不是就唯美了很多了呢？希望你們喜歡，欲知後事如何，歡迎微博關注@大數據分析實戰，我們下回再見。
推薦閱讀：