1. 程式人生 > >wordcloud + lyrics生成歌詞詞雲

wordcloud + lyrics生成歌詞詞雲

根據影象生成歌詞詞雲

class wordcloud.WordCloud(font_path=None, width=400, height=200, margin=2, ranks_only=None, prefer_horizontal=0.9,mask=None, scale=1, color_func=None, max_words=200, min_font_size=4, stopwords=None, random_state=None,background_color='black', max_font_size=None, font_step=1, mode='RGB', relative_scaling=0.5, regexp=None, collocations=True,colormap=None, normalize_plurals=True)


引數
font_path : string //字型路徑,需要展現什麼字型就把該字型路徑+字尾名寫上,如:font_path = '黑體.ttf’如果不指定字型中文字的顯示不出來
width : int (default=400) //輸出的畫布寬度,預設為400畫素
height : int (default=200) //輸出的畫布高度,預設為200畫素
prefer_horizontal : float (default=0.90) //詞語水平方向排版出現的頻率,預設 0.9 (所以詞語垂直方向排版出現頻率為 0.1 )
mask : nd-array or None (default=None) //如果引數為空,則使用二維遮罩繪製詞雲。如果 mask 非空,設定的寬高值將被忽略,遮罩形狀被 mask 取代。
scale : float (default=1)
//按照比例進行放大畫布,如設定為1.5,則長和寬都是原來畫布的1.5倍。
min_font_size : int (default=4) //顯示的最小的字型大小
font_step : int (default=1) //字型步長,如果步長大於1,會加快運算但是可能導致結果出現較大的誤差。
max_words : number (default=200) //要顯示的詞的最大個數
stopwords : set of strings or None //設定需要遮蔽的詞,如果為空,則使用內建的STOPWORDS
background_color : color value (default=”black”) //背景顏色,如background_color=‘white’,背景顏色為白色。
max_font_size : int or None (default=None)
//顯示的最大的字型大小
mode : string (default=”RGB”) //當引數為“RGBA”並且background_color不為空時,背景為透明。
relative_scaling : float (default=.5) //詞頻和字型大小的關聯性
color_func : callable, default=None //生成新顏色的函式,如果為空,則使用 self.color_func
regexp : string or None (optional) //使用正則表示式分隔輸入的文字
collocations : bool, default=True //是否包括兩個詞的搭配
colormap : string or matplotlib colormap, default=”viridis” //給每個單詞隨機分配顏色,若指定color_func,則忽略該方法。

from wordcloud import WordCloud
import matplotlib.pyplot as plt

# lyrics
text = open("lyrics.txt",'r', encoding='UTF-8').read()
# background
bg_pic = plt.imread("idol_huge.jpg")
# word
wordcloud = WordCloud(
    font_path="xingkai.TTF",
    mask=bg_pic,
    scale=2,
    background_color="white", 
    max_words=300, 
    max_font_size=50).generate(text)
    
wordcloud.to_file('lyrics_wordcloud.jpg')

偶像老胡

在這裡插入圖片描述
歌詞詞雲
在這裡插入圖片描述

具體實現參考 github/dreamhomes